CODEWITHSUNDEEP
pythonarraysnumpy
f.ashouri
f.ashouri

Reputation: 5667

change the values of an array from nan to zero

I have an array A on python that has some nan values created by numpy.nan. I want to set all the nan values to zero using A[A==numpy.nan] = 0. It doesn't change the array at all. Why is that?

Upvotes: 4

Views: 6463

Answers (2)

ybdesire
ybdesire

Reputation: 1683

You can use the numpy function to change all the nan value at matrix x to zero.

numpy.nan_to_num(x)

Upvotes: 2

mgilson
mgilson

Reputation: 309831

You want np.isnan:

A[np.isnan(A)] = 0

The problem with your code is that (according to IEEE), nan doesn't equal anything -- even itself.

As a side note, there's also

  • np.isinf -> (+/- infinity)
  • np.isfinite -> (not infinity or NaN)
  • np.isposinf -> (+ infinity)
  • np.isneginf -> (- infinity)

Upvotes: 11

Related Questions