计算矩阵中小于某一值的所有值

问题描述

我必须计算矩阵(二维数组)中小于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是一个图像，我正在将其转换为矩阵，然后查找这些值。

有没有更简单的方法？

---

解决方案

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])

具体从哪里获取结果将取决于您希望结果以何种形式出现。