生成器列表输出异常原因及基于生成器表达式的修复方案咨询

我懂你遇到的这个坑——这是生成器表达式延迟求值特性带来的典型问题！咱们先把问题掰明白，再用纯生成器表达式的方式搞定它。

问题根源 #

你的原代码里，生成器表达式 (i + axis_max for i in [2, 3, 4]) 并不会立刻计算结果，它只会把计算逻辑“存”起来，直到你调用 list(gen) 时才会真正迭代求值。而这时候外层的 for axis_max in x 循环已经跑完了，axis_max 已经被更新成循环的最后一个值 2，所以两个生成器都用了同一个 axis_max=2 来计算，自然得到了两个相同的结果。

纯生成器表达式的修复方案 #

我们可以在生成器表达式内部，通过一个临时的单元素循环来捕获当前迭代的 axis_max 值，避免后续循环变量被覆盖的问题。修改后的代码如下：

x = [1, 2]
axes = []
for axis_max in x:
    # 用单元素for循环绑定当前的axis_max到局部变量current_max
    axis_grid = (i + current_max for i in [2, 3, 4] for current_max in [axis_max])
    axes.append(axis_grid)
print([list(gen) for gen in axes])

为什么这个方法有效？ #

for current_max in [axis_max] 这个子句会立即执行（因为它迭代的是一个只有当前 axis_max 值的列表），把当前循环中的 axis_max 数值绑定到生成器内部的局部变量 current_max 上。这样，当后续迭代生成器时，每个生成器引用的都是自己捕获的 current_max 值（第一个是1，第二个是2），而不是外层循环的变量 axis_max。

运行修改后的代码，输出就是你预期的：

[[3, 4, 5], [4, 5, 6]]

内容的提问来源于stack exchange，提问作者Anton Bohdanov