在python中将矩阵的值更改为高于阈值

Question

我有一个矩阵：matrix = np.array([[[0,0.5,0.6],[0.9,1.2,0]],[[0,0.5,0.6],[0.9,1.2,0]]])我想将所有值替换0.55 < x < 0.95为0.55.PS：我的问题与这个问题类似。但答案在我的情况下不起作用。

Answer 1

您可以使用np.where：

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

matrix[np.where((matrix > 0.55) & (matrix < 0.95))] = 0.55

# Or

# matrix[(matrix > 0.55) & (matrix < 0.95)] = 0.55

输出：

>>> matrix

array([[[0.  , 0.5 , 0.55],

        [0.55, 1.2 , 0.  ]],

       [[0.  , 0.5 , 0.55],

        [0.55, 1.2 , 0.  ]]])