lambda 函数的列表或生成器在 Python 中不返回相同的结果

Question

当我这样做时：

def create_funcs():
    return (lambda: i for i in range(5))

for f in create_funcs():
    print(f())

我得到了预期：

1
2
3
4

但是当我这样做时：

def create_funcs():
    return [lambda: i for i in range(5)]

for f in create_funcs():
    print(f())

我有点奇怪：

4
4
4
4

谁能解释一下为什么？

Answer 1

您的lambda 函数从周围的作用域捕获变量i。如this question 中所述，范围是理解的范围。

您的循环一次遍历生成器中的项目，并在推进生成器之前调用每个项目。当你得到第一个函数时，生成器在i为0的状态下挂起，所以这就是函数看到的值。当您推进生成器时，它会将i 设置为 1，然后您将获得一个获取该值的函数。生成器“等待”更改i，直到您进入下一个函数。

在列表推导中，您可以一次获得所有功能。但是所有人仍然在理解范围内引用相同的变量i。由于列表推导立即一直到最后，i 在您有机会使用任何函数之前一直到 4。

在这两种情况下，您的 lambda 都引用了理解范围内的变量。只是在生成器理解中，这个变量直到你已经调用了前一个函数之后才会前进。如果您存储第一个函数，您可以看到与列表理解情况类似的行为，但在您再次推进生成器之前不要调用它：

def create_funcs():
    return (lambda: i for i in range(5))

gen = create_funcs()
f1 = next(gen)
f2 = next(gen)
print(f1())
print(f2())

输出是

1
1

由于我改进了生成器，i 为 1，因此 f1 和 f2 在调用它们时都会看到该值。如果你执行for f in list(create_funcs())，你可以看到同样的事情，强制生成器像列表理解一样一直走到最后。

Answer 2

正如 BrenBarn 所暗示的，列表推导中的变量 i 是推导范围的局部变量。如果您使用括号，这将起作用：

def create_funcs():
    return (lambda: i for i in range(5))

for f in create_funcs():
    print(f())

0
1
2
3
4