python - Python 列表中生成器的多种用法

Question

基本上，我处于以下情况 - 我生成一个列表，例如

l = [2*x for x in range(10)]

我稍后会多次迭代，例如

for i in l: print i    # 0,2,4,6,8,10,12,14,16,18
for i in l: print i    # 0,2,4,6,8,10,12,14,16,18
for i in l: print i    # 0,2,4,6,8,10,12,14,16,18

问题是列表太大而无法放入内存，因此我使用它的生成器形式，即:

l = (2*x for x in range(10))

但是，在这个构造之后只有第一次迭代有效:

for i in l: print i    # 0,2,4,6,8,10,12,14,16,18
for i in l: print i    #
for i in l: print i    #

问题出在哪里？我如何多次迭代它？

Answer 1

你的发电机第一次就耗尽了。您应该每次都重新创建生成器以更新它:

l = (2*x for x in range(10))
for i in l: print i
l = (2*x for x in range(10))
for i in l: print i

(注意:您应该在 python 2 中使用 xrange，因为 range 在内存中创建一个列表)

您还可以创建一个快捷功能来帮助您，甚至是 generator function :

def gen():
    for i in range(10):
        yield 2 * i

然后:

 for i in gen(): print i
 for i in gen(): print i