为什么我不能对相同的数据进行两次迭代？

Question

老实说，我在这里有点困惑，为什么我不能对相同的数据进行两次迭代？

def _view(self,dbName):
    db = self.dictDatabases[dbName]
    data = db[3]

    for row in data:
        print("doing this one time")

    for row in data:
        print("doing this two times")

这将打印出“做这一次”几次（因为数据有几行），但它根本不会打印出“做这两次”......

我第一次迭代数据工作正常，但第二次运行最后一个列表“for row in data”时，这没有返回任何内容......所以执行一次有效但不是两次......？

仅供参考 - 数据是一个 csv.reader 对象（如果这是原因）...

Answer 1

这是因为data 是一个迭代器，你只能使用一次迭代器。例如：

lst = [1, 2, 3]
it = iter(lst)

next(it)
=> 1
next(it)
=> 2
next(it)
=> 3
next(it)
=> StopIteration

如果我们使用for 循环遍历一些数据，最后一个StopIteration 将导致它第一次退出。如果我们尝试再次对其进行迭代，我们将不断收到StopIteration 异常，因为迭代器已被使用。

现在是第二个问题：如果我们确实需要多次遍历迭代器怎么办？一个简单的解决方案是创建一个包含元素的列表，我们可以根据需要多次遍历它。只要列表中的元素很少，就可以了：

data = list(db[3])

但如果有很多元素，最好使用tee() 创建独立的迭代器：

import itertools
it1, it2 = itertools.tee(db[3], n=2) # create as many as needed

现在我们可以依次循环遍历每一个：

for e in it1:
    print("doing this one time")

for e in it2:
    print("doing this two times")

Answer 2

我想为那些在2017年寻找解决方案并使用python 2.7或3的人完成@ÓscarLópez的答案。

方法 tee() 现在不接受关键字参数，并等待第二个参数是整数，而不是关键字。这才是tee()的正确使用方式：

import itertools
it1, it2 = itertools.tee(db[3], 2)

Answer 3

一旦迭代器用完，它就不会再产生了。

>>> it = iter([3, 1, 2])
>>> for x in it: print(x)
...
3
1
2
>>> for x in it: print(x)
...
>>>