python numpy / scipy在多维数组中查找相对变量的计数或频率

时间:2013-03-13 21:57:36

标签: python numpy scipy

我有一个三维数组,比如dat.shape =(100,128,256)。我正在尝试计算第一个轴上值大于10.0的周期数。例如,对于dat [:,0,0],大于10.0的值出现多少次?然后,dat [:,0,1]到dat [:,n,m]。我的结束矩阵的形状为(128,156)。

有没有办法在numpy或scipy中进行这种计算而不会在第一维和第二维上循环?

非常感谢!

1 个答案:

答案 0 :(得分:3)

import numpy as np
a = np.random.randint(0, 100, (100,128,256))
np.sum(a > 10, axis=0)