python - 如何克隆 Python 生成器对象？

Question

考虑这种情况:

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os

walk = os.walk('/home')

for root, dirs, files in walk:
    for pathname in dirs+files:
        print os.path.join(root, pathname)

for root, dirs, files in walk:
    for pathname in dirs+files:
        print os.path.join(root, pathname)

我知道这个例子有点多余，但你应该考虑到我们需要多次使用相同的 walk 数据。我有一个基准场景，并且必须使用相同的 walk 数据才能获得有用的结果。

我已经尝试 walk2 = walk 来克隆并在第二次迭代中使用，但没有成功。问题是......我怎样才能复制它？有可能吗？

提前谢谢你。

Answer 1

您可以使用 itertools.tee() :

walk, walk2 = itertools.tee(walk)

请注意，正如文档指出的那样，这可能“需要大量额外的存储空间”。