Python numpy 统计数组某一元素的个数/ 统计数组满足条件的元素个数

我们在数据处理的时候，有时需要统计数组中某个元素的个数，如果写个循环就太麻烦了，numpy中有相应的函数进行统计。

1 np.sum() 进行统计

np.sum() 有个特性，当数组为[True,False]时会累加True的个数
所以当我们需要计算数组array中值value的个数时，使用语句：np.sum(data == value) 即可
当我们需要计算数组array中大于值value的个数时，使用语句：np.sum(data > value) 即可

实例1 统计等于某元素的个数：

import numpy as np

data=np.array([[1,2,3],[0,1,2],[3,4,5]])
print("查看数组中各位置是否为1 \n",data == 1)

n = np.sum(data == 1)
print("数组中1的个数",n)

输出结果：

查看数组中各位置是否为1 
 [[ True False False]
 [False  True False]
 [False False False]]
数组中1的个数 2

实例2 统计满足条件的元素的个数

import numpy as np

data=np.array([[1,2,3],[0,1,2],[3,4,5]])
n = np.sum(data > 1)
print("数组中元素大于1的个数为",n)

结果

数组中元素大于1的个数为 6

参考文献

python 统计数组中某个元素的个数