1. ホーム
  2. python

[解決済み] numpy配列のNaN値のインデックスを取得する方法は?

2023-01-07 08:36:10

質問

例えば、次のように定義されたnumpyの配列があるとします。

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

ここで、欠損値のインデックスをすべて含むリストが欲しいのですが、これは [(1,2),(2,0)] である。

何かいい方法はないでしょうか?

どのように解決するのですか?

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]])