在 Python 中将 for 循环转换为矢量化形式答案

【问题标题】：Converting for-loop into vectorized form in Python在 Python 中将 for 循环转换为矢量化形式
【发布时间】：2023-02-05 16:00:00
【问题描述】：

我试图用矢量化形式替换下面的 for 循环，但出现错误。我如何解决它？

for i in range(0,len(sigma[0])):
             if(sigma[0][i]<=threshold): 
                    sigma[0][i]=threshold

我使用的矢量化形式是

sigma[sigma <= threshold] = threshold

我得到的错误

in <module>
    sigma[sigma <= threshold] = threshold

TypeError: '<=' not supported between instances of 'list' and 'float'

【问题讨论】：

【解决方案1】：

在没有额外依赖的情况下，你能做的最好的事情就是列表理解：

sigma[0] = [max(threshold, x) for x in sigma[0]]

更好的是，如 cmets 中所建议的那样，为此使用 Numpy（这将操作应用于整个数组，就目前而言）：

import numpy as np
sigma_arr = np.array(sigma)

sigma_arr[sigma_arr <= threshold] = threshold

或者：

sigma_arr.clip(min=threshold, out=sigma_arr)

【讨论】：