We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f6d9deb commit 75e3466Copy full SHA for 75e3466
1 file changed
flox/aggregate_flox.py
@@ -118,5 +118,7 @@ def nanmean(group_idx, array, *, axis=-1, size=None, fill_value=None, dtype=None
118
masked = np.where(mask, 0, array)
119
out = sum(group_idx, masked, size=size, axis=axis, dtype=dtype, fill_value=fill_value)
120
with np.errstate(invalid="ignore", divide="ignore"):
121
- out /= sum(group_idx, mask.view(np.int8), size=size, axis=axis, fill_value=0, dtype=np.intp)
+ out /= sum(
122
+ group_idx, (~mask).view(np.int8), size=size, axis=axis, fill_value=0, dtype=np.intp
123
+ )
124
return out
0 commit comments