python 统计数组中大于某个值的数的个数

import numpy as np

a = np.random.randint(-5, 5, (1, 10))

c=np.sum(a>=1) #条件为大于等于1

print ("随机数组a : "+str(a)) #输出数组a

print ("大于等于1的个数: "+str(c)) #输出满足条件的个数

np.sum((log_prob>-0.2).cpu().numpy().astype(int)) > 0  #把tensor 转为numpy,之后再统计array中大于-0.2的元素的个数

posted @ 2021-10-08 15:46  呦呦南山  阅读(4726)  评论(0编辑  收藏  举报