diff options
| author | Ethan Kruse <eakruse@uw.edu> | 2015-10-19 13:29:01 -0700 |
|---|---|---|
| committer | Ethan Kruse <eakruse@uw.edu> | 2015-10-21 15:46:49 -0700 |
| commit | 5caf4c932e43c47d73fad761e3257bb0d4551cc2 (patch) | |
| tree | 963b87fbb2b55b64bc41e0148a6fbb1f9d5bde26 /numpy/lib/function_base.py | |
| parent | 17146b2a873e835630ad58d00b73e97ba1fdb214 (diff) | |
| download | numpy-5caf4c932e43c47d73fad761e3257bb0d4551cc2.tar.gz | |
BUG: Make median work for empty arrays (issue #6462)
np.median([]) returns NaN. Fixes bug/regression that raised an IndexError.
Added tests to ensure continued support of empty arrays.
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 555d08386..fef69dff3 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -3339,7 +3339,7 @@ def _median(a, axis=None, out=None, overwrite_input=False): indexer[axis] = slice(index-1, index+1) # Check if the array contains any nan's - if np.issubdtype(a.dtype, np.inexact): + if np.issubdtype(a.dtype, np.inexact) and sz > 0: # warn and return nans like mean would rout = mean(part[indexer], axis=axis, out=out) part = np.rollaxis(part, axis, part.ndim) |