替换所有高于阈值的 RGB 值

我有一个充满 RGB 值的 numpy 3d 数组，例如 形状 = (高度,宽度,3)

  matrix = np.array( [[[0,0.5,0.6],[0.9,1.2,0]])

如果任何值高于阈值，我必须替换 RGB 值，例如阈值 = 0.8，替换 = [2,2,2] 然后

matrix = [[[0,0.5,0.6],[2,2,2]]

我怎样才能用 numpy 有效地做到这一点？ 目前我正在使用双 for 循环并检查是否有任何 RGB 值高于阈值，我将其替换，但是对于 n = 4000 数组来说这非常慢。

我如何使用 numpy 更有效地做到这一点，也许使用 np.where ？

我已经用另一个矩阵扩展了你的矩阵width方面。

matrix = np.array([[[0,0.5,0.6],[0.9,1.2,0]],[[0,0.5,0.6],[0.9,1.2,0]]])

您可以使用以下方法构建掩码np.any在轴 2 上（从 0 开始，因此是第三轴）：

mask = np.any((matrix > 0.8), axis=2)

# mask:
array([[False,  True],
       [False,  True]], dtype=bool)

matrix[mask] = np.array([2,2,2])

您的结果matrix:

array([[[ 0. ,  0.5,  0.6],
        [ 2. ,  2. ,  2. ]],

       [[ 0. ,  0.5,  0.6],
        [ 2. ,  2. ,  2. ]]])