python - izip 内的 izip : unpacking generators?

Question

我有一个 list lg发电机数量，lg = [g1, g2, ..., gn] 。这些生成器被合并为一个生成器 izip , G = izip(*lg) 。 lg 中的发电机之一(例如 g1 )通常通过 izip 自行构建，因此在迭代 G 时，其生成的结果显示为元组。但是，我需要解压元组，即在迭代 G 时, G应该生成一个没有嵌套元组的元组。 构建时是否可以解压g1 G ？

编辑:示例(@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)

Answer 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