From ee42d25974471f5873ff05ad13519405a5a5b49d Mon Sep 17 00:00:00 2001 From: skwbc Date: Sun, 7 Aug 2016 06:54:33 +0900 Subject: BUG: modified ndim condition to use masked_array --- numpy/ma/extras.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/ma/extras.py') diff --git a/numpy/ma/extras.py b/numpy/ma/extras.py index c5c33a2ad..cbf7b6cdb 100644 --- a/numpy/ma/extras.py +++ b/numpy/ma/extras.py @@ -630,7 +630,7 @@ def median(a, axis=None, out=None, overwrite_input=False, keepdims=False): m = np.median(getdata(a, subok=True), axis=axis, out=out, overwrite_input=overwrite_input, keepdims=keepdims) - if isinstance(m, np.ndarray) and 2 <= m.ndim: + if isinstance(m, np.ndarray) and 1 <= m.ndim: return masked_array(m, copy=False) else: return m -- cgit v1.2.1