Как найти nan в numpy?

@cloyd 

Вы можете использовать функцию numpy.isnan(), чтобы найти значения NaN в массиве. Например:

1
2
3
4

import numpy as np

arr = np.array([1, 2, np.nan, 4, 5])
print(np.isnan(arr))

Это выведет массив из булевых значений, где True соответствует значению NaN в исходном массиве.

1

[False False  True False False]

Вы можете также использовать метод np.isnan() для массива, как в следующем примере:

1
2
3
4

import numpy as np

arr = np.array([1, 2, np.nan, 4, 5])
print(arr[np.isnan(arr)])

Это выведет только значения NaN в массиве

1

[nan]