NumPy配列内のゼロに等しい要素のインデックスを見つける 質問する

NumPy配列内のゼロに等しい要素のインデックスを見つける 質問する

NumPyには効率的な関数/メソッドがあるnonzero()オブジェクト内のゼロ以外の要素のインデックスを識別するndarray。要素のインデックスを取得する最も効率的な方法は何ですか?する値はゼロですか?

ベストアンサー1

numpy.where()は私のお気に入り。

>>> x = numpy.array([1,0,2,0,3,0,4,5,6,7,8])
>>> numpy.where(x == 0)[0]
array([1, 3, 5])

このメソッドはwhere、それぞれが入力の異なる次元に対応する ndarray のタプルを返します。入力は 1 次元なので、[0]タプルの唯一の要素をアンボックスします。

おすすめ記事