为什么生成器列表只返回最后一个生成器的元素?

问题描述

我得到了一个任意的对象列表(例如 ['foo','bar'])。我的目标是生成一个大小相等的列表,其中结果列表中的每个元素都是一个将相应输入元素重复 5 次的生成器。

这是我真正想做的事情的一个很大的简化,我知道有很多方法可以解决这个任务。 然而,我在如何解决这个问题时偶然发现了一些我无法解释的奇怪行为。

这是我对上述任务的解决方案:

my_iterators = [
    (element for _ in range(5))
    for element in ["foo","bar"]
]
for my_iterator in my_iterators:
    print(list(my_iterator))

我现在预计输出是:

['foo','foo','foo']
['bar','bar','bar']

然而,令我惊讶的是:

['bar','bar']
['bar','bar']

为什么 (element for _ in range(5)) 似乎是 input_list 中最后一个元素的迭代器,而不管它在 for element in ["foo","bar"] 的上下文中实际上是什么? 我需要如何调整我的代码以实现我最初的目标?

解决方法

正如 jonsharpe 和 David Buck 在评论中指出的那样,实际上导致此处出现非直观行为的是 Python's late binding

似乎有多种方法可以修复已发布的代码,其中一种是:

def make_iterator(element):
    return (element for _ in range(5))

my_iterators = [
    make_iterator(element)
    for element in ["foo","bar"]
]
for my_iterator in my_iterators:
    print(list(my_iterator))