Bug fixes for Dataset.reduce() and n-dimensional cumsum/cumprod #2156
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Fixes GH1470, "Dataset.mean drops coordinates"
Fixes a bug where non-scalar data-variables that did not include the
aggregated dimension were not properly reduced:
Previously::
>>> ds = Dataset({'x': ('a', [2, 2]), 'y': 2, 'z': ('b', [2])})
>>> ds.var('a')
<xarray.Dataset>
Dimensions: (b: 1)
Dimensions without coordinates: b
Data variables:
x float64 0.0
y float64 0.0
z (b) int64 2
Now::
>>> ds.var('a')
<xarray.Dataset>
Dimensions: (b: 1)
Dimensions without coordinates: b
Data variables:
x int64 0
y int64 0
z (b) int64 0
Finally, adds support for n-dimensional cumsum() and cumprod(), reducing
over all dimensions of an array. (This was necessary as part of the above fix.)
stickler-ci
reviewed
May 17, 2018
xarray/tests/test_duck_array_ops.py
Outdated
| def test_cumprod_2d(): | ||
| inputs = np.array([[1, 2], [3, 4]]) | ||
|
|
||
| expected = np.array([[1, 2], [3, 2*3*4]]) |
Contributor
There was a problem hiding this comment.
E226 missing whitespace around arithmetic operator
jhamman
approved these changes
May 18, 2018
xarray/core/dataset.py
Outdated
| keep_attrs=keep_attrs, | ||
| allow_lazy=allow_lazy, | ||
| **kwargs) | ||
| # drop other data_vars |
Member
Author
There was a problem hiding this comment.
This was intentional, but I don't think it actually adds much value.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Fixes GH1470, "Dataset.mean drops coordinates"
Fixes a bug where non-scalar data-variables that did not include the
aggregated dimension were not properly reduced:
Finally, adds support for n-dimensional cumsum() and cumprod(), reducing
over all dimensions of an array. (This was necessary as part of the above fix.)
whats-new.rstfor all changes andapi.rstfor new API (remove if this change should not be visible to users, e.g., if it is an internal clean-up, or if this is part of a larger project that will be documented later)