热门标签:
Q:

python惯用python for loop if else语句

如何在惯用Pythonfor循环中使用for语句? 没有for我可以写例如:

res = [i for i in [1,2,3,4,5] if i < 4]

结果为:[1, 2, 3]

上述代码的正常形式是:

res = []
for i in [1,2,3,4,5]:
    if i < 4:
        res.append(i)

结果与惯用形式相同:[1, 2, 3]

我想要这个:

res = [i for i in [1,2,3,4,5] if i < 4 else 0]

我得到SyntaxError: invalid syntax。 结果应该是:[1, 2, 3, 0, 0] 这个的正常代码是:

res = []
for i in [1,2,3,4,5]:
    if i < 4:
        res.append(i)
    else:
        res.append(0)

结果为:[1, 2, 3, 0, 0]

原网址
A:

你很接近,你只需要将三元移动到你正在创建值的列表理解部分。

res = [i if i < 4 else 0 for i in range(1,6)] 

相似问题