计算矩阵中大于一个值的所有值

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

该numpy.where功能是您的朋友。因为它是为充分利用数组数据类型而实现的，所以对于大图像，您应该注意到与提供的纯python解决方案相比，速度有所提高。

直接使用numpy.where将产生一个布尔掩码，指示某些值是否符合您的条件：

>>> data
array([[1, 8],
       [3, 4]])
>>> numpy.where( data > 3 )
(array([0, 1]), array([1, 1]))

而且掩码可以用于直接索引数组以获取实际值：

>>> data[ numpy.where( data > 3 ) ]
array([8, 4])

从那里获取的确切位置取决于您想要的结果形式。