[Python]Numpy配列の要素のインデックスを条件で検索する(array find index of element by condition)には？

どうも、ちょげ(@chogetarou)です。

Numpy配列の要素のインデックス(index)を条件で検索する方法を紹介します。

方法

Numpy配列の要素のインデックス(index)を条件で検索するには、numpy.where()を使います。

まず、numpy.where()を呼び出します。

numpy.where()の引数に、条件式を指定します。

条件式では、対象のNumpy配列で比較します。

そして、numpy.where()のインデックス「０」にアクセスします。

#condition=要素の条件式
#条件式ではNumpy配列と比較
result = np.where(condition)[0]

上記のnumpy.where()は、条件式を満たすNumpy配列の要素のインデックスを全て取得します。

import numpy as np

nums = np.array([1, 2, 3, 4, 5, 6])

result = np.where(nums % 3 == 0)[0]

print(result)

出力：
[2 5]