我们在数据处理的时候,有时需要统计数组中某个元素的个数,如果写个循环就太麻烦了,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