From c607456d9eac8743e5280adeec5496fd2621ae6a Mon Sep 17 00:00:00 2001 From: Eric Wieser Date: Thu, 27 Apr 2017 23:01:49 +0100 Subject: MAINT: Remove weird create-a-copy dance --- numpy/lib/nanfunctions.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) (limited to 'numpy/lib/nanfunctions.py') diff --git a/numpy/lib/nanfunctions.py b/numpy/lib/nanfunctions.py index c96e925aa..d9df9563f 100644 --- a/numpy/lib/nanfunctions.py +++ b/numpy/lib/nanfunctions.py @@ -61,14 +61,10 @@ def _replace_nan(a, val): NaNs, otherwise return None. """ - is_new = not isinstance(a, np.ndarray) - if is_new: - a = np.array(a) + a = np.array(a, subok=True, copy=True) + if not issubclass(a.dtype.type, np.inexact): return a, None - if not is_new: - # need copy - a = np.array(a, subok=True) mask = np.isnan(a) np.copyto(a, val, where=mask) -- cgit v1.2.1 From a77709c67051a1d19ed4caa83b0b7686ba22bfc4 Mon Sep 17 00:00:00 2001 From: Eric Wieser Date: Thu, 27 Apr 2017 23:02:43 +0100 Subject: BUG: Fix incorrect behavior of nanfunctions on object arrays Fixes gh-8974 and gh-9008 --- numpy/lib/nanfunctions.py | 24 ++++++++++++++++-------- 1 file changed, 16 insertions(+), 8 deletions(-) (limited to 'numpy/lib/nanfunctions.py') diff --git a/numpy/lib/nanfunctions.py b/numpy/lib/nanfunctions.py index d9df9563f..1e342b932 100644 --- a/numpy/lib/nanfunctions.py +++ b/numpy/lib/nanfunctions.py @@ -63,11 +63,17 @@ def _replace_nan(a, val): """ a = np.array(a, subok=True, copy=True) - if not issubclass(a.dtype.type, np.inexact): - return a, None + if a.dtype == np.object_: + # object arrays do not support `isnan` (gh-9009), so make a guess + mask = a != a + elif issubclass(a.dtype.type, np.inexact): + mask = np.isnan(a) + else: + mask = None + + if mask is not None: + np.copyto(a, val, where=mask) - mask = np.isnan(a) - np.copyto(a, val, where=mask) return a, mask @@ -228,8 +234,9 @@ def nanmin(a, axis=None, out=None, keepdims=np._NoValue): kwargs = {} if keepdims is not np._NoValue: kwargs['keepdims'] = keepdims - if not isinstance(a, np.ndarray) or type(a) is np.ndarray: - # Fast, but not safe for subclasses of ndarray + if type(a) is np.ndarray and a.dtype != np.object_: + # Fast, but not safe for subclasses of ndarray, or object arrays, + # which do not implement isnan (gh-9009), or fmin correctly (gh-8975) res = np.fmin.reduce(a, axis=axis, out=out, **kwargs) if np.isnan(res).any(): warnings.warn("All-NaN axis encountered", RuntimeWarning, stacklevel=2) @@ -335,8 +342,9 @@ def nanmax(a, axis=None, out=None, keepdims=np._NoValue): kwargs = {} if keepdims is not np._NoValue: kwargs['keepdims'] = keepdims - if not isinstance(a, np.ndarray) or type(a) is np.ndarray: - # Fast, but not safe for subclasses of ndarray + if type(a) is np.ndarray and a.dtype != np.object_: + # Fast, but not safe for subclasses of ndarray, or object arrays, + # which do not implement isnan (gh-9009), or fmax correctly (gh-8975) res = np.fmax.reduce(a, axis=axis, out=out, **kwargs) if np.isnan(res).any(): warnings.warn("All-NaN slice encountered", RuntimeWarning, stacklevel=2) -- cgit v1.2.1