Python 生成器表达式中的变量范围

Question

我已经编写了一个函数来创建一个字典映射字符串 -> 生成器表达式。生成器表达式根据两个条件过滤项目列表，这两个条件对于字典中的每个生成器都是不同的。

def iters(types):
    iterators = {}
    for tname in types:
        inst, type = tname.split('|')
        iterators[tname] = (t for t in transactions() if t['institution_type'] == inst and t['type'] == type)
    return iterators

我遇到的问题是所有生成器都根据inst 和type 的最后一个值进行过滤，大概是因为这两个变量在循环的每次迭代中都被重复使用。我该如何解决这个问题？

Answer 1

是的，inst 和 type 名称用作 闭包；当您迭代生成器时，它们已绑定到循环中的最后一个值。

为名称创建一个新范围；一个函数可以做到这一点：

def iters(types):
    def build_gen(tname):
        inst, type = tname.split('|')
        return (t for t in transactions()
                if t['institution_type'] == inst and t['type'] == type)
    iterators = {}
    for tname in types:
        iterators[tname] = build_gen(tname)
    return iterators

你也可以用字典理解替换最后几行：

def iters(types):
    def build_gen(tname):
        inst, type = tname.split('|')
        return (t for t in transactions() 
                if t['institution_type'] == inst and t['type'] == type)
    return {tname: build_gen(tname) for tname in types}