numpy数组中大于0的元素个数
可以使用numpy的sum函数和布尔索引来实现:
import numpy as np
arr = np.array([[-1, 2, 0], [3, -4, 5]])
count = np.sum(arr > 0)
print(count) # 输出:3
首先,我们创建了一个2x3的numpy数组arr,其中包含6个元素。然后,我们使用布尔索引arr > 0来创建一个与arr相同形状的布尔型数组,其中大于0的元素为True,其他为False。最后,我们使用np.sum函数对这个布尔型数组进行求和,得到大于0的元素个数。
原文地址: https://www.cveoy.top/t/topic/yyy 著作权归作者所有。请勿转载和采集!