고성능 파이썬(2판)

131

5

장

이터레이터와 제너레이터

이 장에서 배울 내용

• 제너레이터에서 메모리를 절약하는 방법

• 제너레이터가 유용한 상황

• 복잡한 제너레이터 작업에

itertools

를 사용하는 방법

• 지연 연산이 효과적인 경우와 그렇지 않은 경우

다른 프로그래밍 언어에 익숙한 사람들이 파이썬을 처음 접하면 파이썬의

for

루프 표기법에

놀라곤 한다. 다른 언어의 루프는 보통 다음과 같은 형태다.

# 다른 언어

for

(i=0; i<N; i++) {

do_work(i);

}

파이썬에서는

range

라는 새로운 함수를 도입했다.

# 파이썬

for

i

in

range(N):

do_work(i)

이터레이터와 제너레이터

CHAPTER

5

132

고성능 파이썬(2판)

이 파이썬 코드는 루프가 지속되는 데 필요한 모든 데이터를 만들어내는

range

라는 함수를 호

출하는 것처럼 보인다. 직관적으로 생각해보면,

1

부터

1

억까지 수를 세려고 항목이

1

억 개인

배열을 만들어야 한다면 시간이 오래 걸리고 메모리도 많이 필요할 것이다. 여기서

제너레이터

generator

가 위력을 발휘한다. 제너레이터를 사용하면 이런 종류의 함수를