diff options
| author | Bas van Beek <b.f.van.beek@vu.nl> | 2021-05-22 16:04:26 +0200 |
|---|---|---|
| committer | Bas van Beek <b.f.van.beek@vu.nl> | 2021-05-22 16:04:47 +0200 |
| commit | a8b825c0379972234a86f30b76ae9fc853a88b5e (patch) | |
| tree | 84d4355fa29ce3835c882cb011f95c676dd0b640 /numpy/lib/nanfunctions.py | |
| parent | f91f4bcd050299c930092390b54ce9ba51fd70e0 (diff) | |
| download | numpy-a8b825c0379972234a86f30b76ae9fc853a88b5e.tar.gz | |
BUG: Fixed an issue wherein `_nanmedian_small` would fail for `timedelta64`-based dtypes
Diffstat (limited to 'numpy/lib/nanfunctions.py')
| -rw-r--r-- | numpy/lib/nanfunctions.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/numpy/lib/nanfunctions.py b/numpy/lib/nanfunctions.py index 02ad01a98..719c529c1 100644 --- a/numpy/lib/nanfunctions.py +++ b/numpy/lib/nanfunctions.py @@ -1010,10 +1010,12 @@ def _nanmedian_small(a, axis=None, out=None, overwrite_input=False): for i in range(np.count_nonzero(m.mask.ravel())): warnings.warn("All-NaN slice encountered", RuntimeWarning, stacklevel=4) + + fill_value = np.timedelta64("NaT") if m.dtype.kind == "m" else np.nan if out is not None: - out[...] = m.filled(np.nan) + out[...] = m.filled(fill_value) return out - return m.filled(np.nan) + return m.filled(fill_value) def _nanmedian_dispatcher( |