generator와 yield 키워드란 무엇인가요?

Question

Accepted Answer

**generator**는 `return` 대신 `yield` 키워드를 사용하여 값을 **지연적으로, 한 번에 하나씩** 생성하는 함수입니다. 전체 결과를 메모리에 구축하는 대신 각 값을 필요할 때 계산합니다 — 대용량 또는 무한 시퀀스에 이상적입니다.

## yield vs return

```python
# 일반 함수는 전체 리스트를 메모리에 구축함
def get_squares(n):
    return [i ** 2 for i in range(n)]   # n개의 값이 한 번에 생성됨

# generator는 값을 한 번에 하나씩, 필요할 때 yield함
def gen_squares(n):
    for i in range(n):
        yield i ** 2                     # 여기서 일시 정지, 한 값을 반환, next()에서 재개

for sq in gen_squares(1_000_000):        # 거의 메모리를 사용하지 않음
    print(sq)
```

`yield`는 함수를 **일시 정지**하고 값을 반환하며, 다음 반복 시 그 정확한 지점에서 **재개**합니다 — 값들 사이에서 지역 상태를 보존합니다. 요청되기 전까지는 아무것도 계산되지 않습니다.

## 메모리 이점

```python
sum(i ** 2 for i in range(10_000_000))   # generator expression — 일정한 메모리
sum([i ** 2 for i in range(10_000_000)]) # list — 먼저 거대한 리스트를 구축 (낭비)
```

대용량 데이터의 경우 generator는 메모리를 일정하게 유지하지만, list comprehension은 모든 것을 한 번에 구체화합니다.

## 무한 시퀀스 (list로는 불가능)

```python
def count_up():
    n = 0
    while True:        # 무한 — 하지만 지연적이므로 괜찮음
        yield n
        n += 1

gen = count_up()
next(gen)   # 0
next(gen)   # 1  — 요청할 때만 계산
```

값이 필요할 때 생성되기 때문에 generator는 *무한* 스트림을 표현할 수 있습니다 — 필요한 만큼만 가져오면 됩니다.

## 내부적으로 반복이 동작하는 방식

```python
gen = gen_squares(3)
next(gen)   # 0  — 첫 yield까지 실행
next(gen)   # 1  — 재개하여 다음 yield까지 실행
next(gen)   # 4
next(gen)   # StopIteration — generator 소진됨
```

## generator expression

```python
(x ** 2 for x in range(10))   # list comprehension과 비슷하지만 지연적 (괄호)
```

## 왜 중요한가

generator는 대용량 또는 스트리밍 데이터의 **메모리 효율적인** 처리에 필수적입니다 — 거대한 파일을 한 줄씩 읽기, 큰 데이터셋 처리, 파이프라인, 무한 시퀀스 등 전체 리스트를 구축하면 메모리가 고갈될 상황에서요.

`yield`(상태가 보존된 일시 정지/재개)와 지연 평가를 이해하면 확장 가능한 데이터 처리 코드를 작성할 수 있고, generator expression은 동일한 이점을 간결하게 제공합니다.

데이터가 메모리에 담기에 너무 크거나, 모든 것을 미리 처리하기보다 생성되는 대로 항목을 처리하고 싶을 때마다 핵심 도구입니다.