[Python] Generator 합치기

https://stackoverflow.com/questions/3211041/how-to-join-two-generators-or-other-iterables-in-python

How to join two generators (or other iterables) in Python?

 

Python Generator는 '+' 연산을 지원하지 않는다.

 

만약 두 개의 Generator를 합치려는 시도를 '+'를 사용하여 하려한다면 아래와 같은 Error를 보게될것이다.

 

TypeError: unsupported operand type(s) for +: 'generator' and 'generator'

 

'+' 대신 itertools의 chain을 활용하면 두개의 generator를 쉽게 합칠 수 있다.

from itertools import chain

def generator1():
    for item in 'abcdef':
        yield item

def generator2():
    for item in '123456':
        yield item

generator3 = chain(generator1(), generator2())
for item in generator3:
    print(item)