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

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

该

numpy.where

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

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

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

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

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

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