[英]Does python have a built-in function for interleaving generators/sequences?
我注意到itertools
没有(在我看来)有一个函数能够交错来自其他几个可迭代对象的元素(而不是压缩它们):
def leaf(*args): return (it.next() for it in cycle(imap(chain,args)))
tuple(leaf(['Johann', 'Sebastian', 'Bach'], repeat(' '))) => ('Johann', ' ', 'Sebastian', ' ', 'Bach', ' ')
(编辑)我问的原因是因为我想避免不必要的拉链/展平事件。
显然, leaf
的定义很简单,但如果有一个预定义的函数做同样的事情,我宁愿使用它,或者一个非常清晰的生成器表达式。 是否有内置的函数,在itertools中,或在其他一些着名的库中,或者是一个合适的惯用语表达式?
编辑2:可以使用更简洁的定义(使用functional
包):
from itertools import *
from functional import *
compose_mult = partial(reduce, compose)
leaf = compose_mult((partial(imap, next), cycle, partial(imap, chain), lambda *args: args))
您正在寻找内置的zip
和itertools.chain.from_iterable
来展平结果:
>>> import itertools
>>> list(zip(['Johann', 'Sebastian', 'Bach'], itertools.repeat(' ')))
[('Johann', ' '), ('Sebastian', ' '), ('Bach', ' ')]
>>> list(itertools.chain.from_iterable(_))
['Johann', ' ', 'Sebastian', ' ', 'Bach', ' ']
请注意,我使用list
只是为了强制输出一个好的。 使用标准的itertools, leaf
替代实现将是:
leaf = lambda *a: itertools.chain.from_iterable(itertools.izip(*a)) # Python 2.x
leaf = lambda *a: itertools.chain.from_iterable(zip(*a)) # Python 3.x
itertools roundrobin()配方将是我的第一选择,尽管在你的确切示例中它将产生无限序列,因为它以最长的可迭代而不是最短的顺序停止。 当然,解决这个问题很容易。 也许值得检查一下不同的方法?
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.