Python 3中如何重复使用zip迭代器？

你遇到的这个现象是Python迭代器的核心特性导致的：zip()返回的对象是一次性迭代器，当你完成第一次遍历后，迭代器内部的指针已经走到了末尾，再次遍历就不会有任何元素返回了。下面给你几种可行的解决方案，你可以根据场景选择：

1. 转换为列表（你当前使用的方法） #

这是最直接的方式，把zip迭代器转换为列表，列表是可重复遍历的容器，会一次性存储所有元素：

L_RANGES = list(zip(range(10, 20), range(11, 21)))

# 第一次遍历
for a, b in L_RANGES:
    print(a, b)

# 第二次遍历依然正常输出
for a, b in L_RANGES:
    print(a, b)

优点是简单直观，支持随机访问；缺点是如果zip生成的元素数量极大，会占用较多内存。

2. 每次需要时重新生成Zip对象 #

如果担心内存占用，可以把生成zip的逻辑封装成函数，每次需要遍历的时候调用函数生成新的zip迭代器：

def create_zip_ranges():
    return zip(range(10, 20), range(11, 21))

# 第一次遍历
for a, b in create_zip_ranges():
    print(a, b)

# 第二次遍历调用函数生成新对象，正常输出
for a, b in create_zip_ranges():
    print(a, b)

这种方式不会提前存储所有元素，内存效率更高，适合处理大数据量的场景。

3. 使用itertools.tee复制迭代器 #

如果已经有一个现成的zip迭代器，想要复制出多个可独立遍历的迭代器，可以用itertools.tee工具：

import itertools

L_RANGES = zip(range(10, 20), range(11, 21))
# 复制为两个独立的迭代器
zip_iter1, zip_iter2 = itertools.tee(L_RANGES)

# 遍历第一个迭代器
for a, b in zip_iter1:
    print(a, b)

# 遍历第二个迭代器，依然有输出
for a, b in zip_iter2:
    print(a, b)

需要注意的是：如果其中一个复制后的迭代器比另一个遍历得更远，tee会缓存中间元素，此时内存占用可能比直接转列表还高，所以这个方法更适合需要同时或先后遍历几次，但不想提前存储所有元素的场景。

内容的提问来源于stack exchange，提问作者torayeff