我注意到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))