python - 同一个迭代器的多个迭代器(使用枚举)，这是怎么回事？

Question

考虑以下示例:

s = 'abc'
[(c1, c2) for j, c2 in enumerate(s) for i, c1 in enumerate(s)]

输出:

[('a', 'a'),
 ('b', 'a'),
 ('c', 'a'),
 ('a', 'b'),
 ('b', 'b'),
 ('c', 'b'),
 ('a', 'c'),
 ('b', 'c'),
 ('c', 'c')]

如果在列表推导式之外调用 enumerate 并且将迭代器分配给变量，我希望得到相同的输出:

it1, it2 = enumerate(s), enumerate(s)
[(c1, c2) for j, c2 in it1 for i, c1 in it2]

但我得到:

[('a', 'a'), ('b', 'a'), ('c', 'a')]

到底是怎么回事？我使用 Python 3.6.9。

Answer 1

发生的事情是内部迭代器在外部迭代器的第一次迭代后耗尽:

s = 'abc'
it1 = enumerate(s)
it2 = enumerate(s)

for i, x in it1:
    for j, y in it2:  # <-- gets consumed when i = 0 and stays empty
        ...

相比之下:

s = 'abc'

for i, x in enumerate(s):
    for j, y in enumerate(s):  # <-- gets recreated at each iteration
        ....

如果您需要持久性，请将其包含在 list 中或 tuple :

itr = list(enumerate(s))
print([(c1, c2) for j, c2 in itr for i, c1 in itr])
# [('a', 'a'), ('b', 'a'), ('c', 'a'), ('a', 'b'), ('b', 'b'), ('c', 'b'), ('a', 'c'), ('b', 'c'), ('c', 'c')]

尽管请注意使用 enumerate() 的不同内存占用多次与将其包含在 list 中或 tuple .