Question

编辑：有similar question here处理迭代器重置。然而，下面接受的答案解决了嵌套迭代器的实际问题，并处理了一个容易遗漏的问题，嵌套迭代器不会重置。

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

在下面的示例代码中，我可以看到第二次迭代在与第一次迭代相同的对象上运行，从而产生一个奇怪的结果。将其与下面的C＃进行对比，得出我之后的结果。

有没有办法做我想做的事。我想知道我是否可以复制迭代器或者＃34;检索＆＃34;它来自的功能，但也许有一个更简单的方法。（我知道我可以在下面的玩具示例中拨打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())

Answer 2

itertools.tee从单个iterable创建独立的迭代器。但是，一旦创建了新的iterables，就不应再使用原始的iterable。

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

def PrintCombos(x):
    xx = []
    xx.append(itertools.tee(x))
    n = 0
    for a in xx[0][0]:
        xx.append(itertools.tee(xx[n][1]))
        for b in xx[n+1][0]:
            print('%s - %s' % (a,b));
        n += 1

PrintCombos(MyIter());

Answer 3

我发现使用列表理解来解决这类问题对于获得理想的结果最为有效。

x = [1,2,3,4]
y = [1,2,3,4]

spam = [[s,t] for s in x for t in y]

for x in spam:
    print('%s - %s' %(x[0], x[1]))

输出：

Python - 迭代迭代器两次

3 个答案: