具有多个'if'的python列表理解

Question

我们都知道python的

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

句法.

但是 ,列表推导的AST表示有多个'if'表达式的空间:

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

有人能给我一个python代码的例子,它会产生一个带有多个'if'表达式的AST吗？

Answer 1

语法允许多个if语句,因为您可以在for循环之间混合它们:

[j for i in range(100) if i > 10 for j in range(i) if j < 20]

理解组件应该被视为嵌套语句,上面翻译为:

lst = []
for i in range(100):
    if i > 10:
        for j in range(i):
            if j < 20:
                lst.append(j)

这也意味着您可以使用多个if语句而不需要for循环:

[i for i in range(100) if i > 10 if i < 20]

虽然非敏感(只是将那些使用and或与链式运算符组合在一起),但它确实转换为合法的嵌套语句集:

lst = []
for i in range(100):
    if i > 10:
        if i < 20:
            lst.append(i)

语法和解析器并没有特别禁止这种用法,就像Python不允许你嵌套if语句一样.

请注意,PEP 202 - 列表理解(将此功能添加到语言中的原始提案文档)实际上包含示例部分中的双重理解:

>>> print [(i, f) for i in nums for f in fruit if f[0] == "P" if i%2 == 1]
[(1, 'Peaches'), (1, 'Pears'), (3, 'Peaches'), (3, 'Pears')]

Answer 2

只需将它们堆叠在一起:

[i for i in range(100) if i > 10 if i < 50]

产生11到49之间的整数.

Answer 3

使用内置all()函数允许您在迭代中放置多个布尔表达式或函数,并坚持您的理解.我认为这是一个非常使用的内置并且它保持高可读性.

>>> [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]

Answer 4

该语言参考提供了有关这个更好的主意:

list_comprehension  ::=  expression list_for
list_for            ::=  "for" target_list "in" old_expression_list [list_iter]
list_iter           ::=  list_for | list_if
list_if             ::=  "if" old_expression [list_iter]

正如您所看到的,列表理解是list_iter在末尾用可选项定义的list_iter.现在,这list_iter可以是列表理解的另一部分或if条件.if条件本身再次以另一个可选项结束list_iter.这对于在同一列表理解中使用可选的if条件链接多个for-parts是必要的.你也可以为它构建一个.. if X if Y if Z部分的事实list_iter只是一个副作用.

因此,虽然不需要单独链接多个if条件的可能性,但它允许以这种方式定义整个语法.