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

Question

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

是否有一种方法可以在numpy或scipy中进行此计算，而无需在第一维和第二维上循环?

非常感谢!

Answer 1

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