1 개요
순회할 수 있는 데이터형.
for과 함께 쓰여 순회용으로 만드는 것. 루프의 작용을 제어하기 위해 쓰이는 함수.
연속적인 값들을 만들어낸다.
return이 아닌, yield를 통해 돌려받는다. 이를 이용해 한 번 호출될 때마다 하나의 값을 뱉어낸다.(즉, 인공지능 등에서 학습용 데이터를 메모리에 모두 올려두는 게 아니라, 필요할 때마다 해당 데이터를 빼낼 수 있게 구성된 것이다.) => 덕분에 적은 메모리 만으로도 만들 수 있다.
(근본적으로 함수와 다를 바 없는데.. 함수와 다른 이름으로 분류되네)
1.1 예
def generator():
list = range(3) # 0에서 3 미만까지.
for i in list:
yield i
mygen = generator() # 제너레이터를 생성하고..
for i in mygen:
print(i)
# 다른 출력법.
for i in range(3):
print(next(mygen)) # next함수로 제너레이터의 다음 값을 뽑아낸다.
0, 1, 2가 출력된다.
제너레이터에서 추출할 수 있는 값 이상을 뽑으려 하면 에러가 반환된다.
for 뿐만 아니라 while이나, 반복문 없이 다음과 같이 순차적으로 yield를 사용하는 것도 가능하다.
def generator():
yield 1
yield 2
yield 3
2 출력 값이 여러 개인 경우
def generator():
yield 1, 2
위처럼 1개의 제너레이터에서 2개의 출력값이 나오는 경우, 다음과 같이 인덱스를 사용하여 값에 접근할 수 있다.
gen = generator()
next(gen)[0]