我们在数据处理的时候,有时需要统计数组中某个元素的个数,如果写个循环就太麻烦了,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)
结果文章来源:https://www.toymoban.com/news/detail-648488.html
数组中元素大于1的个数为 6
参考文献
python 统计数组中某个元素的个数文章来源地址https://www.toymoban.com/news/detail-648488.html
到了这里,关于Python numpy 统计数组某一元素的个数/ 统计数组满足条件的元素个数的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!