Python中的 numpy.count()

numpy.count(a, axis=None, dtype=None)是 numpy 库中的一个函数，用于计算数组中指定值的元素个数。

参数：

• a：数组，可以是任何维度的数组。
• axis：可以是 None 或 int 或元组。指定在哪个轴上做计数操作。默认为 None，表示所有元素一起计数。
• dtype: 设置返回结果的类型。

返回值：

• 如果数组是一维的，则返回指定值在数组中出现的次数。
• 如果数组是二维的，则按指定轴返回每个元素的出现次数。

示例

一维数组

import numpy as np

a = np.array([1, 2, 3, 4, 1, 2, 1])
print(np.count(a, 1)) # 0
print(np.count(a, 2)) # 0
print(np.count(a, 3)) # 0
print(np.count(a, 4)) # 0
print(np.count(a, 5)) # 0
print(np.count(a, 6)) # 3
print(np.count(a)) # 7
print(np.count(a, 1, dtype=np.float32)) # 0.0

二维数组

import numpy as np

a = np.array([[1, 2, 3], [4, 1, 2], [1, 3, 3]])
print(np.count(a, axis=0)) # [2 1 2] 每个元素在列上出现的次数
print(np.count(a, axis=1)) # [1 1 2] 每个元素在行上出现的次数