python - 如何解压缩迭代器？

Question

给定一个 xys 对列表，将其解压缩为两个列表的 Python 习惯用法是:

xs, ys = zip(*xys)

如果 xys 是一个迭代器，我如何将它解压缩为两个迭代器，而不将所有内容都存储在内存中？

Answer 1

假设您有一些可迭代的对:

a = zip(range(10), range(10))

如果我正确地解释了您的要求，您可以使用 itertools.tee 为第一个和第二个生成独立的迭代器。 :

xs, ys = itertools.tee(a)
xs, ys = (x[0] for x in xs), (y[1] for y in ys)

注意这将在内存中保留您迭代其中一个与另一个之间的“差异”。