Python嵌套生成器管道：无函数调用的实现问题及原因解析

先看问题中的代码示例：

words = ['heald', 'heals', 'heels', 'heild', 'heily', 'heils', 'helly', 'hello', 'hells', 'herls', 'apple', 'bread']  # 示例单词列表
x = (w for w in words if len(w) == 5)

def mk_gen(x, idx, val):
    return (w for w in x if w[idx] == val)

pattern = 'he_l_'
for idx, val in enumerate(pattern):
    if val == '_':
        continue
    x = mk_gen(x, idx, val) # 方法1：正常输出结果
    # x = (w for w in x if w[idx] == val) # 方法2：返回空列表

print(list(x))
# 方法1输出 -> ['heald', 'heals', 'heels', 'heild', 'heily', 'heils', 'helly', 'hello', 'hells', 'herls']
# 方法2输出 -> []

方法2失效的核心原因 #

生成器表达式是延迟求值的，只有当调用list(x)触发迭代时，才会执行里面的过滤逻辑。而方法2里的生成器表达式引用了循环变量idx和val，这两个变量是自由变量——生成器并不会在赋值时保存它们的当前值，而是在实际执行时去读取变量的最新值。

当循环结束后，idx和val会停留在最后一次循环的取值（这里是idx=3，val='l'）。此时所有通过循环赋值的生成器，都会用这一组最终值去过滤，相当于所有过滤条件都变成了w[3] == 'l'，叠加后自然没有符合条件的单词，所以返回空列表。

而方法1的函数调用方式，每次调用mk_gen时，idx和val会作为实参绑定到函数的局部变量上，生成器表达式引用的是这些局部变量，它们的值在函数调用时就已经固定，不会受后续循环变量变化的影响，因此能正确保留每次循环的过滤条件。

符合需求的实现方式（不调用额外函数） #

要在循环中直接用生成器表达式构建管道，关键是把循环变量的当前值绑定到生成器内部，避免引用自由变量。这里提供两种简洁的实现方式：

方式1：用临时元组绑定当前值 #

在生成器表达式里通过一个临时的单元素循环，把当前idx和val的值固化下来：

words = ['heald', 'heals', 'heels', 'heild', 'heily', 'heils', 'helly', 'hello', 'hells', 'herls', 'apple', 'bread']
x = (w for w in words if len(w) == 5)

pattern = 'he_l_'
for idx, val in enumerate(pattern):
    if val == '_':
        continue
    # 用临时元组绑定当前循环的idx和val值
    x = (w for w in x for current_idx, current_val in [(idx, val)] if w[current_idx] == current_val)

print(list(x))
# 输出：['heald', 'heals', 'heels', 'heild', 'heily', 'heils', 'helly', 'hello', 'hells', 'herls']

方式2：用lambda默认参数捕获当前值 #

通过lambda的默认参数特性，把当前idx和val的值绑定到lambda内部：

words = ['heald', 'heals', 'heels', 'heild', 'heily', 'heils', 'helly', 'hello', 'hells', 'herls', 'apple', 'bread']
x = (w for w in words if len(w) == 5)

pattern = 'he_l_'
for idx, val in enumerate(pattern):
    if val == '_':
        continue
    # 用lambda默认参数绑定当前值
    x = (w for w in x if (lambda i=idx, v=val: w[i] == v)())

print(list(x))
# 输出：['heald', 'heals', 'heels', 'heild', 'heily', 'heils', 'helly', 'hello', 'hells', 'herls']

这两种方式都不需要额外定义函数，直接在循环中构建生成器管道，且能正确保留每次循环的过滤条件，得到预期结果。

内容的提问来源于stack exchange，提问作者Anubhav Gupta