[python]if/else in Python's list comprehension

本文主要是介绍[python]if/else in Python's list comprehension，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

问题

目前有这样一个需求，有一个列表a：

a=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

对a中的偶数做加1操作，对奇数做加2操作，并将结果放到b中：

b=[x+1 for x in a if x%2==0 else x+2]

但是提示语法错误：

b=[x+1 for x in a if a%2==0 else x+2]File "<ipython-input-40-210ed8140251>", line 1b=[x+1 for x in a if x%2==0 else x+2]^
SyntaxError: invalid syntax

解决方法

stackoverflow中对该问题有着细致的解答，我将其中精彩的回答整理出来，链接网址为

https://stackoverflow.com/questions/4406389/if-else-in-a-list-comprehension

正确的代码为：

b=[x+1 if x%2==0 else x+2 for x in a]

语法规则为：

[expression1 if condition1 else expression2 for item in list]

翻译成普通语句为：

for item in list:if condition1:expression1else:expression2

如果不带else语句，list comprehension的语法规则为：

[expression for item in list if condition]

翻译成普通语句为：

for item in list:if condition:expression

这篇关于[python]if/else in Python's list comprehension的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---