Czy python ma wbudowaną funkcję do przeplatania generatorów/sekwencji?

Question

zauważyłem, że itertools nie robi (wydaje mi się) mają funkcję zdolną elementów przeplatania z kilkoma innymi iterowalny obiektów (w przeciwieństwie do kompresowania je):

def leaf(*args): return (it.next() for it in cycle(imap(chain,args))) 
tuple(leaf(['Johann', 'Sebastian', 'Bach'], repeat(' '))) => ('Johann', ' ', 'Sebastian', ' ', 'Bach', ' ') 

(Edit) Pytam o to dlatego, Chcę uniknąć niepotrzebnych zip/flatten wystąpień.

Oczywiście, definicja leaf jest dość prosta, ale jeśli istnieje predefiniowana funkcja, która robi to samo, wolałbym użyć tego lub bardzo przejrzystego wyrażenia generatora. Czy istnieje taka funkcja wbudowana w itertools lub w innej dobrze znanej bibliotece lub w odpowiednim idiomatycznym wyrażeniu?

Edit 2: Jeszcze bardziej zwięzła definicja jest możliwe (przy użyciu pakietu 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)) 

Answer 1

W itertools roundrobin() recipe byłby już mój pierwszy wybór, choć w swoim dokładnym przykład byłoby produkować nieskończoną sekwencję, ponieważ zatrzymuje się przy najdłuższej iterable, nie najkrótszy. Oczywiście łatwo to naprawić. Może warto sprawdzić inne podejście?

Answer 2

szukasz wbudowanej zip i itertools.chain.from_iterable spłaszczyć Rezultat:

>>> import itertools 
>>> list(zip(['Johann', 'Sebastian', 'Bach'], itertools.repeat(' '))) 
[('Johann', ' '), ('Sebastian', ' '), ('Bach', ' ')] 
>>> list(itertools.chain.from_iterable(_)) 
['Johann', ' ', 'Sebastian', ' ', 'Bach', ' '] 

Zauważ, że użyłem list tylko po to, aby wymusić dobre wyjście. Używając standardowych itertools alternatywne implementacje dla leaf byłoby:

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