중간에 두 개의 발전기를 병합하는 방법은 무엇입니까?

Question

두 개의 다른 생성기를 병합하려면 어떻게해야합니까? 모든 반복마다 다른 생성기가 출력량을 얻는 방식으로?

>>> gen = merge_generators_in_between("ABCD","12") 
>>> for val in gen: 
...  print val 
A 
1 
B 
2 
C 
D 

어떻게하면됩니까? 나는 itertools에서 그 기능을 찾지 못했습니다.itertools recipes에서 라운드 로빈에서

Answer 1

봐 :

>>> from itertools import cycle, islice 
>>> def roundrobin(*iterables): 
     "roundrobin('ABC', 'D', 'EF') --> A D E B F C" 
     # Recipe credited to George Sakkis 
     pending = len(iterables) 
     nexts = cycle(iter(it).next for it in iterables) 
     while pending: 
      try: 
       for next in nexts: 
        yield next() 
      except StopIteration: 
       pending -= 1 
       nexts = cycle(islice(nexts, pending)) 


>>> for x in roundrobin("ABCD", "12"): 
     print x 


A 
1 
B 
2 
C 
D