Python——迭代一个迭代器两次

Question

编辑：这里有一个类似的问题涉及迭代器重置。然而，下面接受的答案解决了嵌套迭代器的实际问题，并处理了一个容易错过的问题，即嵌套迭代器不会重置。

有没有办法在 python 中迭代迭代器两次？

在下面的示例代码中，我可以看到第二次迭代与第一次迭代在同一对象上运行，因此产生了奇怪的结果。将此与下面的 C# 进行对比，得到我想要的结果。

有什么办法可以做我想做的事。我想知道是否可以复制迭代器或“检索”它来自的函数，但也许有一种更简单的方法。（我知道我可以在下面的玩具示例中调用MyIter()两次，但如果我不知道迭代器来自哪里并且不是我想要的，那么这是没有用的！）。

def MyIter():
  yield 1;
  yield 2;
  yield 3;
  yield 4;

def PrintCombos(x):
  for a in x:
      for b in x:
          print(a,"-",b);

PrintCombos(MyIter());

给出

1 - 2
1 - 3
1 - 4

对比：

static IEnumerable MyIter()
{
    yield return 1;
    yield return 2;
    yield return 3;
    yield return 4;
}

static void PrintCombos(IEnumerable x)
{
    foreach (var a in x)
        foreach (var b in x)
            Console.WriteLine(a + "-" + b);
}

public static void Main(String[] args)
{
    PrintCombos(MyIter());
}

这使：

1-1
1-2
1-3
1-4
2-1
2-2
. . .

Answer 1

您可以用来itertools.tee创建生成器的多个副本

from itertools import tee

def MyIter():
    yield 1
    yield 2
    yield 3
    yield 4

def PrintCombos(x):
    it1, it2 = tee(x, 2)
    for a in it1:
        it2, it3 = tee(it2, 2)
        for b in it3:
        print("{0}-{1}".format(a, b))

PrintCombos(MyIter())