Question

它更多地是关于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

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

Answer 1

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。

Answer 2

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

Answer 3

只是另一种解决方案，希望有人可能喜欢它：

使用：[False，True] [表达式]

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

Answer 4

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

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

Answer 5

我能够做到这一点

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

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

5 个答案: