Python循环计算时保证r[i]取值大于等于0的实现问题

你尝试的max(x,0) for x in r[i]逻辑触发报错的核心原因是：r[i]是单个数值变量，并非可迭代对象，无法被遍历处理。

方案1：赋值后单独处理r[i]（改动最小，可读性最高） #

直接在解包赋值完成后，对r[i]做非负截断即可：

# 注意如果i从0开始，i-1=-1会取列表末尾元素，建议根据初始化逻辑调整起始值
for i in range(1, N): 
    a[i], b[i], r[i] = [
        A[t] * a[i-1] + B[t] * b[i-1] + C[t] * r[i-1]
        for t in range(3)
    ]
    # 强制r[i]不小于0
    r[i] = max(r[i], 0)

也可以替换为三元表达式实现相同效果：r[i] = r[i] if r[i] >= 0 else 0

方案2：在列表推导阶段就处理r对应的计算结果 #

如果不想额外加一行赋值逻辑，也可以在生成计算列表时单独对第三个（对应r的）结果做处理：

for i in range(1, N):
    calc_res = [A[t] * a[i-1] + B[t] * b[i-1] + C[t] * r[i-1] for t in range(3)]
    a[i], b[i], r[i] = calc_res[0], calc_res[1], max(calc_res[2], 0)

原代码中range(N)会让i从0开始遍历，此时i=0时i-1=-1会取到a/b/r列表的最后一个元素，如果你已经提前初始化了a[0]/b[0]/r[0]的初始值，建议把循环范围改为range(1, N)避免逻辑错误。

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