使用多个'if'的Python列表推导

Question

我们都知道Python的

[f(x) for x in y if g(x)]

语法。

然而 AST列表理解的表示可以容纳多个“if”表达式：

comprehension = (expr target, expr iter, expr* ifs)

有人能给我一个python代码的例子吗？这个例子可以生成一个AST，它有多个if表达式？

Answer 1

使用内置的all()可以将多个布尔表达式或函数放在一个iterable中并坚持理解。我认为这是一个相当少的使用内置，它保持可读性高。

>>> [x for x in range(20) if all([1 < x < 10, not x & 1])]
[2, 4, 6, 8]

或者

>>> [x for x in range(20) if all([foo(x), bar(x)])]

如果只需要满足一个条件，内置的any()也可以在这里正常工作：

>>> [x for x in range(20) if any([1 < x < 10, not x & 1])]
[0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 14, 16, 18]