我有一个三维数组,比如dat.shape =(100,128,256)。我正在尝试计算第一个轴上值大于10.0的周期数。例如,对于dat [:,0,0],大于10.0的值出现多少次?然后,dat [:,0,1]到dat [:,n,m]。我的结束矩阵的形状为(128,156)。
有没有办法在numpy或scipy中进行这种计算而不会在第一维和第二维上循环?
非常感谢!
答案 0 :(得分:3)
import numpy as np
a = np.random.randint(0, 100, (100,128,256))
np.sum(a > 10, axis=0)