🏄🏻 파이썬 제너레이터

이전에 제너레이터를 아주 잠깐 정리하고 넘어갔는데 제대로 공부하고자 다시 정리하려고 합니다.

제너레이터

제너레이터 : 이터레이터를 생성해주는 함수이며, 파이썬의 시퀀스를 생성하는 객체입니다.(제너레이터는 yield 문을 사용합니다)
이터레이터 : 반복문을 활용해서 데이터를 순회하면서 처리하는 것
(리스트, 딕셔너리, 튜플 등)
시퀀스 : 앞뒤가 꼭 지켜져야 하는 시/공간적 순서

1. 제너레이터 없는 for문

def square_numbers(nums):
    result = []
    for i in nums:
        result.append(i * i)
    return result

my_nums = square_numbers([1, 2, 3, 4, 5])

print(my_nums)
>> [1, 4, 9, 16, 25]

print를 실행해보면 위의 함수가 전부 실행되어 값이 반환된 것을 볼 수 있습니다.

2. 제너레이터를 사용한 for문 

def square_numbers(nums):
    for i in nums:
        yield i * i

my_nums = square_numbers([1, 2, 3, 4, 5])  #1

print(my_nums)
>> <generator object gensquare_numbers at 0x7fc5fe928f90>

위의 내용을 프린트해보면 값이 나오는 것이 아닌, 제너레이터 오브젝트가 리턴이 됩니다.
자신이 리턴할 모든 값을 메모리에 저장하는 것이 아닌 호출 될 때마다 하나씩 값을 전달하기 때문에 기다리는 상태입니다.

def square_numbers(nums):
    for i in nums:
        yield i * i

my_nums = square_numbers([1, 2, 3, 4, 5])  #1

# 그냥 프린트가 아닌 next로 감싸서 프린트 해보면 1이 찍혀서 나옵니다.
print(next(my_nums))
>> 1
print(next(my_nums))
>> 4
print(next(my_nums))
>> 9
print(next(my_nums))
>> 16
print(next(my_nums))
>> 25

프린트를 계속 반복해보면 값이 나오는 것을 알 수 있습니다.
값이 전부 반환된 상태에서 한번 더 next 를 프린트 하면,

print(next(my_nums))
>> Traceback (most recent call last):
  File "generator.py", line 22, in 
    print(next(my_nums))
StopIteration

StopIteration 예외가 발생합니다.

제너레이터는 일반적으로 for문을 통해서 호출합니다.

def square_numbers(nums):
    for i in nums:
        yield i * i

my_nums = square_numbers([1, 2, 3, 4, 5])

for num in my_nums:
    print(num)

위의 코드를 실행해보면 예외처리가 발생하지 않고, 알아서 멈춥니다.

데이터의 양이 많아지고 메모리 사용량을 최적화 해야하는 상황이 온다면 제너레이터를 사용해야합니다.

yield 를 사용하지 않고, 제너레이터를 만들 수 있습니다.

result = (i*i for i in range(0,10))
for num in result:
    print(num)

리스트 컴프리헨션 구조에서 괄호를 '[]' 가 아닌 '()' 로 바꿔주면 됩니다.

출처