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的元素的个数