如何在 numpy 数组中找到 nan 值的坐标

Question

我尝试了np.where但没有成功：

>>>a = np.array([np.nan, 1])
>>>np.where(a == np.nan)
(array([], dtype=int64),)

Answer 1

你需要改变

np.where(a == np.nan)

自

np.where(np.isnan(a))

NaN 值在相等性检查中始终返回 false，即使使用另一个 NaN 值也是如此。所以你需要使用特殊函数来检查NaN，比如np.isnan。

Answer 2

import numpy as np
x = np.array([0,0,-1,1,np.nan, 0, 1, np.nan])
print np.where(np.isnan(x))

返回：

(array([4, 7]),)