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


177

更多有关python列表理解语法的信息。我有一个列表推导,它产生给定范围的奇数列表:

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

这将构成一个过滤器-我有一个源列表,其中删除了偶数(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的python表达式:

1 if 0 is 0 else 3

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


1
毫无疑问,您拥有的是“列表理解”,而不是生成器。最终语法是相同的,只是生成器使用()而不是[]
mgilson


2
我花了一段时间才知道为什么要if x % 2 消除偶数(而不是保留它们)–这是因为什么时候x该偶数x % 2表达式结果为0,该表达式的结果为False,而int除以外的任何结果0True

Answers:


327

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

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

混淆是由于您在第一个示例中使用了过滤器而在第二个示例中没有使用过滤器。在第二个示例中,您仅映射使用三元运算符表达式每个值到另一个。

使用过滤器,您需要:

[ EXP for x in seq if COND ]

没有过滤器,您需要:

[ EXP for x in seq ]

在第二个示例中,该表达式是一个“复杂”表达式,其中恰好包含一个if-else


2
我有一个问题…… [x for x in range(1, 10) if x % 2]语法正确。[x if x % 2 for x in range(1, 10)]-这不是,但[x if x%2 else x*100 for x in range(1, 10)]还是正确的语法。怎么会?
ducin 2013年

@tkoomzaaskz在您的第​​二个示例中,它既不是三元运算符(缺少else),也不是过滤器(因为它EXP属于列表理解)
shx2 2015年

3
@tkoomzaaskz要进一步说明,请注意,您可以if在末尾添加第二个:[x if x%2 else x*100 for x in range(1, 10) if not x%3]第一个if是三元运算符的if一部分,第二个是列表理解语法的一部分。整体x if x%2 else x*100就像一个简单的“处于同一水平” 2*x,它是已经for过滤时在左侧评估的表达式if not x%3
zx81 2015年

您好,单行语句的性能要比for i in x:在for循环中的两行中使之更好if i == y:吗?
Alexis.Rolland



10

只是另一种解决方案,希望有人会喜欢它:

使用:[假,真] [表达式]

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

3

我能够做到这一点

>>> [x if x % 2 != 0 else x * 100 for x in range(1,10)]
    [1, 200, 3, 400, 5, 600, 7, 800, 9]
>>>
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.