Python为列表子元素指定阈值时遇语法错误，求修正得预期输出

我尝试生成列表CI，给子列表的每个元素设置阈值threshold——当元素值小于阈值时取阈值，但代码运行时出现语法错误，相关信息如下：

原代码 #

import numpy as np
km = [1, 2, 3, 4, 5]
CI=[]
threshold=0.005

for t in range(0, 2):
    CI_t = []
    for i in range(0, len(km)):
        CI1 = 0.001 * (1 - np.exp(-km[i] * t))
        CI_t=threshold if CI_t < threshold
        CI_t.append(CI1)
    CI.append(CI_t)
print("CI =",CI)

报错信息 #

CI_t=threshold if CI_t < threshold[i]
                                         ^
SyntaxError: invalid syntax

预期输出 #

CI = [[0.0, 0.0, 0.0, 0.0, 0.0], [0.005, 0.005, 0.005, 0.005, 0.005]]

错误原因及修复方案 #

1. 三元表达式语法错误 #

Python的三元表达式必须完整：值1 if 条件 else 值2，你漏写了else分支；而且错误地拿列表CI_t和阈值比较，应该用计算出来的单个元素CI1和阈值对比。

2. 逻辑错误 #

你原本想把处理后的单个元素添加到CI_t列表，但错误地直接给整个列表CI_t赋值阈值，正确逻辑是先处理CI1，再把处理后的值追加到列表。

修正后的代码 #

import numpy as np
km = [1, 2, 3, 4, 5]
CI = []
threshold = 0.005

for t in range(0, 2):
    CI_t = []
    for i in range(len(km)):
        CI1 = 0.001 * (1 - np.exp(-km[i] * t))
        # 对单个元素CI1做阈值处理，再添加到列表
        processed_val = threshold if CI1 < threshold else CI1
        CI_t.append(processed_val)
    CI.append(CI_t)
print("CI =", CI)

运行这段代码就能得到你想要的预期输出。

可选优化：用列表推导式简化代码 #

如果想让代码更简洁，可改用嵌套列表推导式：

import numpy as np
km = [1, 2, 3, 4, 5]
threshold = 0.005

CI = [
    [threshold if (0.001 * (1 - np.exp(-k * t))) < threshold else (0.001 * (1 - np.exp(-k * t))) for k in km]
    for t in range(2)
]
print("CI =", CI)

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