在izip中使用izip:解压缩生成器?

时间:2012-02-13 20:03:49

标签: python generator

我有lg个生成器列表lg = [g1, g2, ..., gn]。这些生成器与izipG = izip(*lg)合并为一个生成器。 lg中的一个生成器(比如g1)通常是通过izip构建的,因此在迭代G时生成的结果显示为元组。但是,我需要解压缩元组,即当迭代G时,G应该生成一个没有嵌套元组的元组。 构建 G时是否可以解压缩g1

编辑:示例(来自@jsbueno的短版本):

from itertools import izip

g1 = izip(xrange(1,3), xrange(3,5))
g2 = xrange(10, 13)
g3 = xrange(20, 23)

lg = [g1, g2, g3]

for i in izip(*lg):
    print i

产生

((1, 3), 10, 20)
((2, 4), 11, 21)

但我需要

(1, 3, 10, 20)
(2, 4, 11, 21)

1 个答案:

答案 0 :(得分:1)

因为对于你想要的东西,“izip”需要有关内部izip内部迭代器的信息 - 而且izip相对于它是不透明的(也就是说,在给定izip对象的情况下,你无法获取izip的参数) - 第一种方法是重新实现izip的功能,但是以一种了解它的方式 - 如果传递的迭代器之一产生的结果是一个元组,将它连接到结果,而不是将其添加为单项:

class FlatIzip(object):
    def __init__(self, *args):
        self.generators = args
    def __iter__(self):
        iters = [iter(x) for x in self.generators]
        while True:
            result = []
            for iterator in iters:
                element = iterator.next()
                if isinstance(element, tuple):
                    result += list(element)
                else:
                    result.append(element)
            yield tuple(result)

#Example:

from itertools import izip

g1 = xrange(10)
g2 = xrange(10,20)
g3 = xrange(20, 30)
g4 = izip(xrange(40,50), xrange(50,60))


lg = [g1, g2, g3, g4]

for x in FlatIzip(*lg):
    print x