finding index of none and nan in numpy array

Question

I have an array which looks like:

array([[  1.,   2.,   None],
       [ nan,   4.,   5.]])

I am trying the following:

np.equal(A, None) #works and finds index of None correctly
np.equal(A, np.nan) #doesn't work
np.isnan(A) #errors out

The error is:

TypeError: ufunc 'isnan' not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''

How should I approach this, I am trying to find the index of None and nan in a given array.

My final output should look as:

array([[False, False,  True],
       [True, False, False]], dtype=bool)

np.nan is a proper float, but None is an object. So your array has dtype object. — hpaulj
– hpaulj, Commented Mar 14, 2017 at 20:21

kennytm · Accepted Answer · 2017-03-14 19:55:10Z

6

We could first cast the array to have dtype float — which will convert the None to NaN. numpy.isnan can then be used on this float array.

numpy.isnan(A.astype(float))

answered Mar 14, 2017 at 19:55

kennytm

526k110 gold badges1.1k silver badges1k bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

finding index of none and nan in numpy array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related