잔재미코딩

파이썬 제너레이터 (Generator)

파이썬을 활용하여 자동화 프로그램 만들기

혹시 자료가 도움이 되셨다면, 제가 수년간의 강의경험으로 기존과 다르게 특별히 만든 파이썬 강의를 익혀보세요!

업무 시간 단축! 파이썬을 활용하여 자동화 프로그램 만들기

1:1 코칭으로 단순 반복 작업을 없애줄 각자만의 자동화 프로그램까지 가져가기

자세히 알아보기

21. 파이썬 제너레이터 (Generator)

  • iterator를 만들어주는 기능
  • generator를 이해하기 위해 yield 키워드를 이해하자

21.1. yield 를 통해 generator를 이해하자

def str_data():
    data = "hi Dave"
    for item in data:
        return item

# str_data 는 함수
str_data

# str_data() 를 호출하면 함수가 실행되고 종료된다.
str_data()

# str_data() 함수를 기반으로 iterator 만들기
char = iter(str_data())
파이썬을 활용하여 자동화 프로그램 만들기

혹시 자료가 도움이 되셨다면, 제가 수년간의 강의경험으로 기존과 다르게 특별히 만든 파이썬 강의를 익혀보세요!

업무 시간 단축! 파이썬을 활용하여 자동화 프로그램 만들기

1:1 코칭으로 단순 반복 작업을 없애줄 각자만의 자동화 프로그램까지 가져가기

자세히 알아보기 
# char 는 iterator
char

next(char)

next(char)
생각해보기1
1. 위 코드에서 에러가 나는 이유를 생각해보자

이미 첫 next() 호출시, iterator인 char는 끝나버렸다!

yield란?

  • return 은 결과값을 리턴하고, 함수를 종료
  • yield 는 yield라고 적은 곳에서 잠시 함수 실행을 멈추고 호출한 곳에 값을 전달
    • 해당 함수는 현재 실행된 상태를 계속 유지
    • 따라서, 다시 해당 함수를 호출하면 현재 실행된 상태를 기반으로 다음 코드를 실행함
def str_data():
    data = "hi Dave"
    for item in data:
        yield item
파이썬을 활용하여 자동화 프로그램 만들기

혹시 자료가 도움이 되셨다면, 제가 수년간의 강의경험으로 기존과 다르게 특별히 만든 파이썬 강의를 익혀보세요!

업무 시간 단축! 파이썬을 활용하여 자동화 프로그램 만들기

1:1 코칭으로 단순 반복 작업을 없애줄 각자만의 자동화 프로그램까지 가져가기

자세히 알아보기 
# str_data 는 함수
str_data

# str_data() 를 호출하면 generator 객체가 생성된다.
str_data()

# str_data() 함수를 기반으로 iterator 만들기
char = iter(str_data())

next(char)

next(char)

print (next(char))
print (next(char))
print (next(char))
생각해보기2
1. 위 코드에서 이번에는 에러가 안나는 이유를 생각해보자
파이썬을 활용하여 자동화 프로그램 만들기

혹시 자료가 도움이 되셨다면, 제가 수년간의 강의경험으로 기존과 다르게 특별히 만든 파이썬 강의를 익혀보세요!

업무 시간 단축! 파이썬을 활용하여 자동화 프로그램 만들기

1:1 코칭으로 단순 반복 작업을 없애줄 각자만의 자동화 프로그램까지 가져가기

자세히 알아보기

iterator인 char는 현재 실행된 상태를 계속 유지하기 때문에 다음 next() 호출시, 현재 상태 기반 다음 코드를 진행

21.2. Generator Expression

  • Generator Comprehension 이라고도 함
  • List Comprehension과 형식 유사
    • List Comprehension은 앞뒤를 [ ] 대괄호로 표현
    • Generator Expression ( ) 둥근 괄호로 표현
  • 실제 컬렉션 데이터를 리턴하지 않고, 표현식만 유지
    • yield 방식으로 실제 호출할 때 관련 데이터 리턴(Lazy Operation)
# list comprehension 예
square_data = [num ** 2 for num in range(5)]

print( type(square_data))
print (square_data)
print (sum(square_data))

print (sum(square_data))
print (sum(square_data))
print (sum(square_data))

# generator expression 예1
square_data = (num ** 2 for num in range(5))

print (type(square_data))
print (square_data)
print (sum(square_data))   # <--- 실제 사용될 때 관련 데이터 리턴(Lazy Operation)
파이썬을 활용하여 자동화 프로그램 만들기

혹시 자료가 도움이 되셨다면, 제가 수년간의 강의경험으로 기존과 다르게 특별히 만든 파이썬 강의를 익혀보세요!

업무 시간 단축! 파이썬을 활용하여 자동화 프로그램 만들기

1:1 코칭으로 단순 반복 작업을 없애줄 각자만의 자동화 프로그램까지 가져가기

자세히 알아보기 
# 위 sum() 함수에서 제너레이터 모든 데이터를 다 리턴하였으므로, 더이상 리턴할 데이터가 없음
print (sum(square_data))

# generator expression 예3
square_data = (num ** 2 for num in range(5))

# 두 개의 데이터만 순회
for num in range(2):
    print (next(square_data))

# 나머지 데이터를 순회
for num in square_data:
    print (num)

for num in lst:
    print(num, end=' ')
    
print()
    
for num in lst:
    print(num, end=' ')

파이썬 Comprehension 그래프 이해