如何避免理解中的 Python 赋值表达式泄漏

Question

在Effective Python一书中，作者建议使用赋值表达式来避免理解中的冗余，例如：

def fun(i):
    return 2 * i

result = {x: y for x in [0, 1, 2, 3] if (y := fun(x)) > 3}

而不是

result = {x: fun(x) for x in [0, 1, 2, 3] if fun(x) > 3}

result 的值为 {2: 4, 3: 6}。

作者说

如果一个推导在推导的值部分使用海象运算符并且没有条件，它会将循环变量泄漏到包含范围中。 [...] 最好不要泄漏循环变量，因此我建议仅在推导式的条件部分使用赋值表达式。

但是，在上面的示例中，y 在程序结束时设置为 6。所以，赋值表达式中的变量泄露了，虽然它是在条件中定义的。

列表推导也会发生同样的事情：

>>> _ = [(x, leak) for x in range(4) if (leak := 2 * x) > 3]
>>> leak
6

甚至对于生成器表达式：

>>> it = ((x, leak) for x in range(4) if (leak := 2 * x) > 3)
>>> next(it)
(2, 4)
>>> leak
4
>>> next(it)
(3, 6)
>>> leak
6

我错过了什么？有什么办法可以完全避免在理解中的赋值表达式泄漏？

Answer 1

在 Python 中，不不可能泄漏循环变量

与 C 或 Java 等其他语言不同，Python 在 if 和 for 块中没有单独的范围。因此，当您在 if 语句、for 循环或列表推导中使用 := 运算符时，分配的变量将 在函数或类定义的其余部分中。这也意味着在每个for 循环之后，循环变量仍将在作用域内并包含循环最后一次迭代的值。

如果Effective Python的作者认为这是一件坏事，我不同意他的看法。 “泄漏”循环变量可能非常有用！考虑以下示例：

while line := f.readLine():
    if 'Kilian' in line:
        break

print('This is the first line that contains your name: ', line)

但是，此规则有一个例外：在列表推导中进行的隐式赋值有自己的范围：

>>> [x for x in range(10)]
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> x
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'x' is not defined

这个例外可能是您的困惑的根源。这只是一种特殊情况，不适用于在列表推导中使用 := 时。