[Python]numpy.count_nonzero()でNumpy配列の特定の要素の数をカウントするには？

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

numpy.count_nonzero()を使って、Numpy配列で特定の要素の数をカウントする方法を紹介します。

方法

numpy.count_nonzero()を使って、Numpy配列で特定の要素の数をカウントするには、条件式を使います。

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

そして、numpy.count_nonzero()の引数に、「Numpy配列 == 要素」のような条件式を指定します。

result = np.count_nonzero(array == item)

上記のnumpy.count_nonzero()は、「==」の左辺の配列で、「==」の右辺の要素の数をカウントします。

使用例

import numpy as np

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

result1 = np.count_nonzero(nums == 1)
result2 = np.count_nonzero(nums == 2)

print(result1) #4
print(result2) #2