[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 1
    b=[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:
        expression1
    else:
        expression2

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

[expression for item in list if condition]

翻译成普通语句为：

for item in list:
    if condition:
        expression