[Python]numpy.sum()でNumpy配列の条件を満たす要素の個数を検索するには？

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

numpy.sum()を使って、Numpy配列で条件を満たす要素の個数を検索する方法を紹介します。

方法

numpy.sum()を使って、Numpy配列で条件を満たす要素の個数を検索するには、条件式を使います。

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

そして、numpy.sum()の引数にNumpy配列自体を比較する条件式を記述します。

#条件式ではnumpy配列自体を比較
result = np.sum(条件式)

上記のnumpy.sum()は、条件式で比較したNumpy配列を検索し、条件式とTrueとなる要素の個数を返します。

使用例

import numpy as np

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

result1 = np.sum(nums < 4)
result2 = np.sum(nums % 2 != 0)

print(result1) #3
print(result2) #5