for循环中的复合条件

Question

Python 允许在列表推导中使用“if”条件，例如：

[l for l in lines if l.startswith('example')]

常规的“for”循环中缺少此功能，因此没有：

for line in lines if line.startswith('example'):
    statements

需要评估循环中的条件：

for line in lines:
    if line.startswith('example'):
        statements

或嵌入生成器表达式，如：

for line in [l for l in lines if l.startswith('example')]:
    statements

我的理解正确吗？有没有比我上面列出的更好或更 Pythonic 的方式来实现在 for 循环中添加条件的相同结果？

请注意，选择“行”只是作为示例，任何集合或生成器都可能存在。

Answer 1

一些不错的想法来自其他答案和 cmets，但我认为 this recent discussion on Python-ideas 和 its continuation 是这个问题的最佳答案。

总结一下：这个想法在过去已经讨论过，考虑到这些好处似乎不足以激发语法更改：

• 增加的语言复杂性和对学习曲线的影响

• 所有实现中的技术更改：CPython、Jython、Pypy..

• 过度使用合成器可能导致的奇怪情况

人们似乎高度重视的一点是避免带来类似 Perl 的复杂性，从而损害可维护性。

This message 和 this one 很好地总结了 for 循环中复合 if 语句的可能替代方案（几乎已经出现在此页面中）：

# nested if
for l in lines:
    if l.startswith('example'):
        body

# continue, to put an accent on exceptional case
for l in lines:
    if not l.startswith('example'):
        continue
    body

# hacky way of generator expression
# (better than comprehension as does not store a list)
for l in (l for l in lines if l.startswith('example')):
    body()

# and its named version
def gen(lines):
    return (l for l in lines if l.startswith('example'))
for line in gen(lines):
    body

# functional style
for line in filter(lambda l: l.startswith('example'), lines):
    body()

Answer 2

也许不是 Pythonic，但你可以filter 行。

for line in filter(lambda l: l.startswith('example'), lines):
    print(line)

当然，如果 lambda 对您造成困扰，或者您想要更复杂的过滤，您也可以定义自己的过滤函数。

def my_filter(line):
    return line.startswith('example') or line.startswith('sample')

for line in filter(my_filter, lines):
    print(line)

我会说在循环中包含条件会更好，因为您在遍历行时不会在内存中维护“过滤”列表。

所以，那就是

for line in file:
    if not my_filter(line):
        continue
    # statements

Answer 3

不是缺少这个功能，除了某些特殊情况，我想不出有什么办法可以做到。 (l for l in lines if l.startswith('example')) 是一个生成器对象，l 变量是该对象的本地变量。 for 只能看到生成器的 __next__ 方法返回的内容。

for 非常不同，因为生成器的结果需要绑定到调用者范围内的变量。你可以写

for line in (line for line in lines if l.startswith('example')):
    foo(line)

安全，因为这两个 line 位于不同的范围内。

此外，生成器不必只返回其局部变量。它可以评估任何表达式。你会怎么做？

for line in (foo(line)+'bar' for line in lines if line.startswith('example')):
    statements

假设你有一个列表列表

for l in (l[:] for l in list_of_lists if l):
    l.append('modified')

不应附加到原始列表中。

Answer 4

有没有比我上面列出的更好或更 Pythonic 的方式来实现在 for 循环中添加条件的相同结果？

不，没有，也不应该有； 这就是为什么列表推导首先出现在这里的理由。来自corresponding PEP：

列表推导提供了一种更简洁的方式在当前使用map() 和filter() 和/或嵌套循环的情况下创建列表。

列表推导构成了嵌套for、ifs 的替代方案；为什么你想要一个替代方案？

如果您需要使用 if 和 for，请将其嵌套在其中，如果您不想这样做，则使用列表推导式。 扁平比嵌套更好 但是 可读性很重要;允许if 存在会导致难以视觉解析的长而难看的行。