전문가를 위한 파이썬(2판)

686

4

부

제어 흐름

17

.

5

.

4

제너레이터의 작동 방식

본체 안에

yield

키워드가 있는 함수는 모두 제너레이터 함수다. 제너레이터 함수는 호출되면

제너레이터 객체를 반환한다. 즉, 제너레이터 함수는 제너레이터 팩토리라고 할 수 있다.

TIP

일반 함수와 제너레이터 함수의 유일한 구문 차이는 제너레이터 함수가 본체 안 어디에선가

yield

키

워드를 사용한다는 점밖에 없다. 제너레이터 함수는

def

가 아니라

gen

과 같은 새로운 키워드를 사용해야 한

다고 주장하는 사람도 있었지만, 귀도 반 로섬은 동의하지 않았다. 그는 ‘

PEP

255

–간단한 제너레이터

Simple

Generators

’ 제안서(

https://fpy.li/pep255

)에 자신의 의견을 피력했다.

5

제너레이터의 작동을 잘 보여 주는 간단한 예는 다음과 같다.

6

예제

17-6

숫자 세 개를 생성하는 제너레이터 함수

>>> def gen_123():

... yield 1 ❶

... yield 2

... yield 3

...

>>> gen_123 # doctest: +ELLIPSIS

<function gen_123 at 0x...> ❷

>>> gen_123() # doctest: +ELLIPSIS

<generator object gen_123 at 0x...> ❸

>>> for i in gen_123(): ❹

... ...