我有lg
个生成器列表lg = [g1, g2, ..., gn]
。这些生成器与izip
,G = 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)
答案 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