python - 如何合并两个 python 迭代器？

Question

我有两个迭代器，一个 list 和一个 itertools.count 对象(即一个无限值生成器)。我想将这两者合并到一个结果迭代器中，该迭代器将在两者之间交替产生值:

>>> import itertools
>>> c = itertools.count(1)
>>> items = ['foo', 'bar']
>>> merged = imerge(items, c)  # the mythical "imerge"
>>> merged.next()
'foo'
>>> merged.next()
1
>>> merged.next()
'bar'
>>> merged.next()
2
>>> merged.next()
Traceback (most recent call last):
    ...
StopIteration

最简单、最简洁的方法是什么？

Answer 1

生成器会很好地解决您的问题。

def imerge(a, b):
    for i, j in itertools.izip(a,b):
        yield i
        yield j