오늘의 파이썬 공부: generators with yield

안녕하세요, 오늘은 파이썬의 중요한 기능 중 하나인 ‘generators with yield’에 대해 알아보겠습니다. 이 개념은 대량의 데이터를 처리하거나 메모리 효율성을 높이고자 할 때 매우 유용하게 사용됩니다. 특히, 반복 가능한 객체를 생성하는 방법을 제공하여 대용량 데이터 처리 시 메모리를 아끼는 데 큰 도움을 줍니다. 그럼 이제 ‘generators with yield’의 기본 개념과 활용 방법에 대해 자세히 알아보도록 하겠습니다.

‘generators with yield’ 기본 개념 이해하기

기본 문법 및 구조

‘generators with yield’의 기본 문법은 함수 내에서 yield 키워드를 사용하는 것입니다. 일반 함수와 달리, 제너레이터 함수는 값을 반환하는 대신 yield를 사용해 값을 생성합니다. 제너레이터 함수는 호출될 때마다 실행을 중단하고 yield에 지정된 값을 반환하며, 다음에 호출될 때는 중단된 지점부터 실행을 재개합니다.

def simple_generator():
    yield 1
    yield 2
    yield 3

for value in simple_generator():
    print(value)

위의 예제에서 simple_generator 함수는 1, 2, 3을 순차적으로 생성합니다. for 루프를 통해 각 값을 순차적으로 출력합니다.

 

실용적인 활용 예제

예제 1

첫 번째 예제에서는 대량의 데이터를 처리할 때 제너레이터를 사용하는 방법을 살펴보겠습니다. 예를 들어, 파일의 각 줄을 읽어올 때 제너레이터를 사용할 수 있습니다.

def read_large_file(file_path):
    with open(file_path, 'r') as file:
        for line in file:
            yield line.strip()

for line in read_large_file('large_file.txt'):
    print(line)

이 예제는 대용량 파일의 메모리 사용을 최소화하면서 각 줄을 처리할 수 있게 해줍니다.

예제 2

두 번째 예제에서는 제너레이터를 사용하여 무한 시퀀스를 생성하는 방법을 살펴보겠습니다. 이는 특히 데이터 스트림을 처리할 때 유용합니다.

def infinite_sequence():
    num = 0
    while True:
        yield num
        num += 1

for number in infinite_sequence():
    if number > 10:  # 제한을 두기 위해 조건을 추가
        break
    print(number)

위 예제는 무한 시퀀스를 생성하지만, for 루프 내에서 조건문을 통해 적절히 제어할 수 있습니다.

 

주의사항 및 팁

흔히 하는 실수

제너레이터를 사용할 때 초보자들이 자주 겪는 실수 중 하나는 yield를 return과 혼동하는 것입니다. yield는 함수 실행을 중지하고 값을 반환하며, 이후에 다시 실행을 재개할 수 있습니다.

알아두면 좋은 팁

제너레이터를 사용할 때는 itertools 모듈과 결합하여 더욱 강력한 기능을 구현할 수 있습니다. 예를 들어, itertools.islice()를 사용하면 제너레이터의 특정 범위의 값만 처리할 수 있습니다.

 

결론 및 요약