Python中如何将numpy数组中小于阈值的元素替换为阈值？

问题背景 #

需要将numpy数组sigma中所有小于阈值threshold的元素替换为threshold，但当前代码未达到预期效果。

核心错误 #

代码中循环内的赋值操作写错：使用了比较运算符==（仅判断相等性，不会修改数组），而非赋值运算符=，导致数组元素未被更新。

修复方案1：修正循环逻辑 #

将sigma[i] == threshold改为sigma[i] = threshold即可生效：

import numpy as np

sigma = np.array([[ 0.02109   ],
                  [ 0.01651925],
                  [ 0.02109   ],
                  [ 0.02109   ],
                  [ -0.009   ]])

threshold = 0.010545

for i in range(0, len(sigma)):
    if(sigma[i] <= threshold):
        sigma[i] = threshold  # 修正赋值运算符

print([sigma])

修复方案2：Numpy原生向量化操作（推荐） #

Numpy支持向量化运算，无需手动循环，效率更高，代码更简洁：

方式一：使用np.maximum

直接对数组和阈值取逐元素最大值，自动替换小于阈值的元素：

import numpy as np

sigma = np.array([[ 0.02109   ],
                  [ 0.01651925],
                  [ 0.02109   ],
                  [ 0.02109   ],
                  [ -0.009   ]])

threshold = 0.010545

sigma = np.maximum(sigma, threshold)
print([sigma])

方式二：布尔索引赋值

通过布尔索引定位所有小于阈值的元素，直接批量赋值：

import numpy as np

sigma = np.array([[ 0.02109   ],
                  [ 0.01651925],
                  [ 0.02109   ],
                  [ 0.02109   ],
                  [ -0.009   ]])

threshold = 0.010545

sigma[sigma <= threshold] = threshold
print([sigma])

以上所有方案运行后，都会得到期望输出：

[array([[0.02109   ],
       [0.01651925],
       [0.02109   ],
       [0.02109   ],
       [0.010545  ]])]

内容的提问来源于stack exchange，提问作者Wiz123