如何獲得numpy數組中所有NaN值的索引列表？

Question

說現在我已經被定義爲一個numpy的陣列，

[[1,2,3,4], 
[2,3,NaN,5], 
[NaN,5,2,3]] 

現在我想有一個包含缺失值，這是[(1,2),(2,0)]在這種情況下的所有索引列表。

有什麼辦法可以做到嗎？

Answer 1

np.isnan聯合np.argwhere

x = np.array([[1,2,3,4], 
       [2,3,np.nan,5], 
       [np.nan,5,2,3]]) 
np.argwhere(np.isnan(x)) 

輸出：

array([[1, 2], 
     [2, 0]]) 

Answer 2

可以使用np.where以匹配對應於所述陣列的Nan值和map每個結果的布爾條件以生成tuples列表。

>>>list(map(tuple, np.where(np.isnan(x)))) 
[(1, 2), (2, 0)]