diff options
Diffstat (limited to 'numpy/ma/extras.py')
| -rw-r--r-- | numpy/ma/extras.py | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/numpy/ma/extras.py b/numpy/ma/extras.py index 4e7f8e85e..41bce0f22 100644 --- a/numpy/ma/extras.py +++ b/numpy/ma/extras.py @@ -732,12 +732,8 @@ def median(a, axis=None, out=None, overwrite_input=False, keepdims=False): else: return m - r, k = _ureduce(a, func=_median, axis=axis, out=out, + return _ureduce(a, func=_median, keepdims=keepdims, axis=axis, out=out, overwrite_input=overwrite_input) - if keepdims: - return r.reshape(k) - else: - return r def _median(a, axis=None, out=None, overwrite_input=False): |