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

Question

为什么我不能在同一个 iterator 上迭代两次?

# data is an iterator.

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

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

这会打印 "doing this one time" 几次，因为 data 是非空的。但是，它不打印“这样做两次”。为什么迭代 data 第一次有效，第二次无效？

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 是一个迭代器:

data = list(data)

只要列表中的元素很少就可以。但是，如果有很多元素，最好使用 tee() 创建独立的迭代器。 :

import itertools
it1, it2 = itertools.tee(data, 2) # create as many as needed

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

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

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