diff options
| author | empeeu <empeeu@users.noreply.github.com> | 2015-06-22 19:42:37 -0400 |
|---|---|---|
| committer | empeeu <empeeu@users.noreply.github.com> | 2015-06-22 19:42:37 -0400 |
| commit | 3e82d6db99bc9e32a6f6bcb63cbfccbaa1168a74 (patch) | |
| tree | 3f81f0bbb6ab6fb263e9450ef5dba26e3394af0c /numpy/lib/function_base.py | |
| parent | a320fd772468004a53f7c448ae47032eb1b5c5df (diff) | |
| download | numpy-3e82d6db99bc9e32a6f6bcb63cbfccbaa1168a74.tar.gz | |
Moving warning outside of for loop.
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 762d338bb..d4db4dab1 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -3083,10 +3083,10 @@ def _median(a, axis=None, out=None, overwrite_input=False): rout = out else: rout = a.dtype.type(np.nan) - else: - for i in range(np.count_nonzero(n.ravel())): - warnings.warn("Invalid value encountered in median", - RuntimeWarning) + elif np.count_nonzero(n.ravel()) > 0: + warnings.warn("Invalid value encountered in median for" + + " %d results" % np.count_nonzero(n.ravel()), + RuntimeWarning) rout[n] = np.nan return rout else: |