在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

Answer 2

您也可以直接迭代生成器：

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