혼자 공부하는 파이썬 개정판 66강 - 이터러블, 이터레이터, 제너레이터 함수과 제너레이터 표현식

이터러블과 이터레이터의 개념

파이썬에서 '이터러블'과 '이터레이터'는 코드의 반복 처리를 위한 기본 개념이에요. 이터러블(Iterable)은 반복할 수 있는 객체를 의미하고, 리스트, 튜플, 딕셔너리 등이 이에 해당하죠. 이터레이터(Iterator)는 이터러블을 순회할 수 있는 객체로, 'next()' 함수를 통해 한 번에 하나의 요소를 꺼낼 수 있어요. 이 두 개념을 이해하는 것이 중요해요.

이터러블을 직접 만들고 싶다면 이터레이터를 생성해야 해요. 이터레이터는 크게 세 가지 방법으로 만들 수 있는데, 그 중 두 가지가 제너레이터 표현식과 제너레이터 함수에요. 제너레이터 표현식은 간단하고 직관적으로 이터레이터를 생성할 수 있는 방법이에요. 이를 통해 데이터 처리를 효율적으로 진행할 수 있죠.

제너레이터 표현식

제너레이터 표현식은 리스트 내포와 유사한 방식으로 사용돼요. 리스트 내포는 대괄호를 사용하지만, 제너레이터 표현식은 소괄호를 사용해요. 예를 들어, 1부터 100까지의 숫자를 제곱하는 리스트를 만들고 싶다면 리스트 내포를 사용하면 메모리가 많이 소모되지만, 제너레이터 표현식을 사용하면 메모리 사용량을 줄일 수 있어요.

코드를 작성해보면, gen = (x**2 for x in range(1, 101))와 같이 구현할 수 있어요. 이렇게 생성된 제너레이터는 'next()' 함수를 통해 하나씩 꺼내 사용할 수 있어요. 첫 번째 호출 시에는 1이 나오고, 두 번째 호출 시에는 4가 나오죠. 이처럼 제너레이터 표현식은 메모리를 절약하면서도 필요한 시점에 데이터 처리를 가능하게 해요.

제너레이터 함수와 메모리 효율성

제너레이터 함수는 'yield' 키워드를 사용하여 이터레이터를 생성하는 함수에요. 함수 안에서 'yield'를 사용하면, 해당 함수는 제너레이터로 취급돼요. 예를 들어, def my_generator(): yield x**2와 같은 형태로 정의할 수 있어요. 이 함수를 호출하면 제너레이터 객체가 생성되고, 이 역시 'next()' 함수를 통해 요소를 하나씩 꺼낼 수 있어요.

리스트 내포와 제너레이터 표현식의 가장 큰 차이점은 메모리 사용량이에요. 리스트 내포는 모든 데이터를 한 번에 생성하지만, 제너레이터는 필요한 순간에만 데이터를 생성하죠. 예를 들어, 1억 개의 데이터를 처리할 때 제너레이터를 사용하면 메모리 과부하를 피할 수 있어요. 복잡한 연산을 진행할 때도 제너레이터를 활용하면 효율적으로 작업할 수 있답니다.