이터레이터와 제너레이터를 활용한 데이터 합성

17 Sep 2023

python

데이터 합성은 프로그래밍에서 매우 일반적인 작업입니다. 이터레이터와 제너레이터는 데이터 합성에 효과적으로 사용할 수 있는 도구입니다. 이번 블로그 포스트에서는 이터레이터와 제너레이터를 사용하여 데이터를 합성하는 방법에 대해 알아보겠습니다.

이터레이터란 무엇인가요?

이터레이터는 순서대로 값을 반환하는 객체입니다. 값을 하나씩 순차적으로 처리할 수 있도록 해주는 역할을 합니다. 이터레이터 객체는 __iter__()와 __next__() 메서드를 가져야 하며, __iter__() 메서드는 이터레이터 객체 자신을 반환하고, __next__() 메서드는 다음 값을 반환합니다. 더 이상 반환할 값이 없을 때 StopIteration 예외를 발생시켜 작업을 종료합니다.

아래는 간단한 숫자 이터레이터의 예시입니다.

class NumberIterator:
    def __init__(self, limit):
        self.limit = limit
        self.current = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.current < self.limit:
            self.current += 1
            return self.current
        raise StopIteration

위 코드에서 NumberIterator 클래스는 __iter__()와 __next__() 메서드를 구현하여 이터레이터 객체를 생성합니다. 이터레이터 객체를 사용하면 for 루프를 통해 값을 하나씩 처리할 수 있습니다.

제너레이터란 무엇인가요?

제너레이터는 이터레이터를 간편하게 생성할 수 있는 함수입니다. 일반 함수와는 달리 return 키워드 대신 yield 키워드를 사용하여 값을 반환합니다. yield 키워드를 사용하면 함수의 상태가 유지되고 다음 호출 시 이전 상태에서 진행됩니다. 제너레이터 함수는 이터레이터 객체를 생성하여 값을 하나씩 반환하는 역할을 합니다.

다음은 제너레이터를 사용한 간단한 예시입니다.

def number_generator(limit):
    current = 0
    while current < limit:
        yield current
        current += 1

위 코드에서 number_generator 함수는 제너레이터 객체를 생성하여 값을 반환하는 역할을 합니다. 이를 사용하면 for 루프와 같은 이터레이션 작업에 간편하게 활용할 수 있습니다.

데이터 합성에 이터레이터와 제너레이터 활용하기

이제 이터레이터와 제너레이터를 사용하여 데이터를 합성하는 방법에 대해 알아보겠습니다. 데이터 합성은 여러 개의 데이터 소스를 합쳐서 하나의 데이터 스트림으로 처리하는 작업입니다.

아래는 이터레이터와 제너레이터를 활용한 데이터 합성의 예시입니다.

def merge_iterators(*iterators):
    for it in iterators:
        yield from it

# 숫자 이터레이터 생성
number_iter = NumberIterator(5)

# 문자열 이터레이터 생성
string_iter = iter("Hello")

# 데이터 합성
merged_iter = merge_iterators(number_iter, string_iter)

# 합성된 데이터 처리
for data in merged_iter:
    print(data)

위 코드에서 merge_iterators 함수는 여러 개의 이터레이터를 받아들여 하나의 이터레이터로 합성하는 역할을 합니다. yield from 구문을 사용하여 각 이터레이터에서 값을 반환합니다.

위 예시에서는 숫자 이터레이터와 문자열 이터레이터를 생성하여 데이터를 합성하고, for 루프를 통해 합성된 데이터를 처리합니다. 결과는 숫자와 문자열이 번갈아가며 출력됩니다.

결론

이터레이터와 제너레이터는 데이터 합성에 효과적으로 활용될 수 있는 도구입니다. 이터레이터를 사용하면 값을 하나씩 순차적으로 처리할 수 있으며, 제너레이터를 사용하면 간편하게 이터레이터 객체를 생성하여 데이터를 합성할 수 있습니다. 이를 통해 데이터 처리를 더욱 효율적으로 수행할 수 있습니다.

#python #iterator #generator