"isnotnan" functionality in numpy, can this be more pythonic?

arrays numpy python nan
AnalyticsBuilder picture AnalyticsBuilder · May 14, 2010 · Viewed 78.6k times · Source

I need a function that returns non-NaN values from an array. Currently I am doing it this way:

>>> a = np.array([np.nan, 1, 2])
>>> a
array([ NaN,   1.,   2.])

>>> np.invert(np.isnan(a))
array([False,  True,  True], dtype=bool)

>>> a[np.invert(np.isnan(a))]
array([ 1.,  2.])

Python: 2.6.4 numpy: 1.3.0

Please share if you know a better way, Thank you

Answer

mtrw picture mtrw · May 14, 2010 
a = a[~np.isnan(a)]

Related questions

Removing nan values from an array

I want to figure out how to remove nan values from my array. My array looks something like this: x = [1400, 1500, 1600, nan, nan, nan ,1700] #Not in this exact configuration How can I remove the nan values from x?

Read More
Numpy - Replace a number with NaN

I am looking to replace a number with NaN in numpy and am looking for a function like numpy.nan_to_num, except in reverse. The number is likely to change as different arrays are processed because each can have …

Read More
numpy array: replace nan values with average of columns

I've got a numpy array filled mostly with real numbers, but there is a few nan values in it as well. How can I replace the nans with averages of columns where they are?

Read More