单行列表理解:if-else变体

这是关于Python列表理解语法。 我有一个列表理解产生给定范围的奇数列表:

[x for x in range(1, 10) if x % 2] 

这使得一个filter – 我有一个源列表,我删除偶数( if x % 2 )。 我想在这里使用if-then-else之类的东西。 以下代码失败:

 >>> [x for x in range(1, 10) if x % 2 else x * 100] File "<stdin>", line 1 [x for x in range(1, 10) if x % 2 else x * 100] ^ SyntaxError: invalid syntax 

有一个像if-else一样的pythonexpression式:

 1 if 0 is 0 else 3 

如何在列表理解中使用它?

x if y else z是您为每个元素返回的expression式的语法。 因此你需要:

 [ x if x%2 else x*100 for x in range(1, 10) ] 

混淆起因于你在第一个例子中使用filter ,而不是在第二个例子中。 在第二个例子中,只使用三元运算符expression式每个值映射到另一个值。

使用filter,您需要:

 [ EXP for x in seq if COND ] 

没有filter,你需要:

 [ EXP for x in seq ] 

在你的第二个例子中,expression式是一个“复杂的”,恰好涉及到一个if-else

 [x if x % 2 else x * 100 for x in range(1, 10) ] 

只是另一个解决办法,希望有人会喜欢它:

使用:[False,True] [expression式]

 >>> map(lambda x: [x*100, x][x % 2 != 0], range(1,10)) [1, 200, 3, 400, 5, 600, 7, 800, 9] >>> 

你也可以用列表理解来做到这一点:

  A=[[x*100, x][x % 2 != 0] for x in range(1,11)] print A 

我能够做到这一点

 >>> [x if x % 2 != 0 else x * 100 for x in range(1,10)] [1, 200, 3, 400, 5, 600, 7, 800, 9] >>>