python : 列表推导式中的 if else 的连用

前言

• 之前，我一直认为在python的列表推导式及各种推导式中是只可以有if，不能有else的。
• 直到我看到了 cookbook 这本书中有关于这一点的写法。我又get到一个小技能。
• 顺便说一下：这本书真的很不错。

代码

• 之前，我以为只有三元运算符可以这样使用。

    tmp = 1 if True else 0

• 但是实际上：

In [1]: a=[1,2,3,4,]

In [2]: a
Out[2]: [1, 2, 3, 4]

In [3]: b=[i for i in a if i>2]  # 我一直以为只可以这样

In [4]: b
Out[4]: [3, 4]

In [5]: b=[i for i in a if i>2 else 5] # 因为每当我这样使用的时候就会报错
  File "<ipython-input-5-1a54c5082971>", line 1
    b=[i for i in a if i>2 else 5]
                              ^
SyntaxError: invalid syntax


In [6]: b=[i if i>2 else -1 for i in a] #但实际上这样使用才是对的

In [7]: b
Out[7]: [-1, -1, 3, 4]

In [8]: ['零' if i==0 else '三' if i==3 else '五' if i==5 else i  for i in range(20) ] # 这样看起来骚不骚
Out[8]: ['零', 1, 2, '三', 4, '五', 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]

总结

• 本文仅是作者作为备忘记录的，但是我相信应该存在读者和我有过一样的问题。