关于itertools中zip与tee交互时原迭代器状态差异的问询

核心原因是混淆了**可迭代对象（Iterable）和迭代器（Iterator）**的本质区别，这直接决定了tee的工作方式：

1. 先明确两个关键概念 #

• 可迭代对象：比如range、列表、字符串这类，实现了__iter__方法，每次调用iter()都会生成一个全新的迭代器，可以反复遍历。
• 迭代器：比如zip()、map()的返回值，或是调用iter()后的结果，实现了__next__方法，内部有状态指针，只能被遍历一次，遍历结束就彻底耗尽。

2. 拆解你的两个示例 #

示例1：range(3)的预期行为 #

from itertools import tee

a = range(3)
a1,a2 = tee(a,2)
print(list(a1), list(a2), list(a))
# 输出: [0, 1, 2], [0, 1, 2], [0, 1, 2]

range(3)是可迭代对象而非迭代器。当tee创建a1和a2时，这两个副本在遍历过程中，会各自调用iter(a)生成独立的迭代器读取元素。所以遍历a1、a2完全不会影响原可迭代对象a的状态，后续调用list(a)时，依然能生成新的迭代器拿到完整数据。

示例2：zip(...)的"意外"行为 #

from itertools import tee

a = zip(range(3),range(3,6))
a1,a2 = tee(a,2)
print(list(a1), list(a2), list(a))
# 输出: [(0, 3), (1,4), (2,5) ], [(0, 3), (1, 4), (2,5)], []

zip()返回的是迭代器，本身只能被遍历一次。tee处理迭代器时会共享原迭代器的状态：

• 遍历a1时，tee会不断调用next(a)获取元素，同时把这些元素缓存起来供a2后续读取；
• 当a1遍历完成，原迭代器a的指针已经走到末尾，彻底耗尽；
• 此时再调用list(a)，迭代器没有元素可返回，自然得到空列表。

3. tee的工作逻辑总结 #

• 传入可迭代对象：tee让每个副本独立迭代原对象，互不干扰，原对象状态保持完好。
• 传入迭代器：tee共享原迭代器的状态，所有副本依赖这个迭代器的推进，一旦任意副本耗尽原迭代器，原迭代器本身就无法再读取到内容（不过tee会缓存元素，其他未遍历完的副本仍能拿到缓存数据）。

如果想避免zip这类情况，可以先把迭代器转成可迭代对象（比如列表）：

from itertools import tee

a = list(zip(range(3),range(3,6)))
a1,a2 = tee(a,2)
print(list(a1), list(a2), list(a))
# 输出: [(0, 3), (1,4), (2,5) ], [(0, 3), (1, 4), (2,5)], [(0, 3), (1,4), (2,5)]

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