python - 计算矩阵中小于一个值的所有值

Question

我必须计算矩阵(二维数组)中小于 200 的所有值。

我为此写的代码是:

za=0   
p31 = numpy.asarray(o31)   
for i in range(o31.size[0]):   
    for j in range(o32.size[1]):   
        if p31[i,j]<200:   
            za=za+1   
print za

o31 是一个图像，我将其转换为矩阵，然后找到值。

有没有更简单的方法来做到这一点？

Answer 1

这对于 bool 数组来说非常简单:

p31 = numpy.asarray(o31)
za = (p31 < 200).sum() # p31<200 is a boolean array, so `sum` counts the number of True elements