Python Academy · 강의

반복자 프로토콜

__iter__와 __next__ 이해하기

반복자 프로토콜은(는) CoddyKit의 무료 Python Academy 강의입니다. 이것은 4개 중 1번째 강의입니다. 아래에서 전체 강의를 무료로 읽을 수 있으며, 내장 코드 에디터와 24/7 AI 튜터와 함께 브라우저에서 직접 실습할 수 있습니다. 이 강의는 Python Academy 학습 경로의 일부이며, 진행 상황이 웹과 CoddyKit 앱에 동기화됩니다. Python Academy 강의에는 총 4개의 강의가 포함되어 있습니다.

반복이란 무엇인가요

목록, 문자열 또는 사전에 대해 for 반복문을 작성할 때마다 Python은 내부적으로 반복자 프로토콜을 사용합니다.

반복문으로 순회할 수 있는 객체를 반복 가능한 객체라고 합니다. 실제로 한 번에 하나씩 값을 생성하는 것을 반복자라고 합니다.

for ch in 'abc':
    print(ch)

반복자 얻기와 다음 값 가져오기

두 내장 함수가 반복을 진행합니다.

iter(obj)는 반복 가능한 객체에 반복자를 요청합니다.
next(it)는 반복자에 다음 값을 요청합니다.

더 이상 값이 없으면 next()는 StopIteration을 발생시킵니다.

it = iter([10, 20, 30])
print(next(it))
print(next(it))
print(next(it))

StopIteration

next()를 한 번 너무 많이 호출하면 StopIteration이 발생합니다. for 반복문은 이를 자동으로 처리하고 깔끔하게 종료합니다.

it = iter([1])
print(next(it))
try:
    next(it)
except StopIteration:
    print('No more items')

두 이중 밑줄 메서드

직접 반복자를 만들려면 두 메서드를 구현해야 합니다.

__iter__는 반복자 객체 자체를 반환합니다.
__next__는 다음 값을 반환하거나 StopIteration을 발생시킵니다.

class Count:
    def __init__(self, n):
        self.i = 0
        self.n = n
    def __iter__(self):
        return self
    def __next__(self):
        if self.i >= self.n:
            raise StopIteration
        self.i += 1
        return self.i

for x in Count(3):
    print(x)

반복 가능한 객체와 반복자 비교

반복 가능한 객체에는 __iter__만 필요합니다. 반복자에는 __iter__와 __next__가 모두 필요합니다.

목록은 반복 가능하지만 반복자는 아닙니다. iter()를 호출할 때마다 새로운 반복자를 얻기 때문입니다.

nums = [1, 2, 3]
print(iter(nums) is iter(nums))
it = iter(nums)
print(it is iter(it))

반복자는 한 번만 소진됩니다

반복자는 한 번만 소비할 수 있습니다. 끝에 도달한 뒤에는 계속 비어 있습니다. 다시 반복하려면 새로운 반복자를 만들어야 합니다.

it = iter([1, 2])
print(list(it))
print(list(it))

범위와 비슷한 반복자

간단한 카운터처럼 동작하는 사용자 지정 반복자를 만들어 보겠습니다. 상태가 인스턴스에 저장된다는 점에 주목하시기 바랍니다.

class EvenUpTo:
    def __init__(self, limit):
        self.cur = 0
        self.limit = limit
    def __iter__(self):
        return self
    def __next__(self):
        if self.cur > self.limit:
            raise StopIteration
        value = self.cur
        self.cur += 2
        return value

print(list(EvenUpTo(8)))

반복 가능한 객체와 반복자 분리하기

더 깔끔한 설계에서는 컨테이너와 반복자를 분리하므로 컨테이너를 여러 번 순회할 수 있습니다.

class Bag:
    def __init__(self, items):
        self.items = items
    def __iter__(self):
        return BagIterator(self.items)

class BagIterator:
    def __init__(self, items):
        self.items = items
        self.i = 0
    def __iter__(self):
        return self
    def __next__(self):
        if self.i >= len(self.items):
            raise StopIteration
        v = self.items[self.i]
        self.i += 1
        return v

b = Bag(['x', 'y'])
print(list(b))
print(list(b))

기본값과 함께 다음 값 가져오기

반복자가 비어 있을 때 next(it, default)는 StopIteration을 발생시키는 대신 기본값을 반환합니다. 안전하게 다음 값을 미리 확인할 때 유용합니다.

it = iter([42])
print(next(it, 'done'))
print(next(it, 'done'))

메모리 효율적인 반복자

반복자는 한 번에 하나의 항목을 생성하므로 전체 시퀀스를 메모리에 보관할 필요가 없습니다. 따라서 매우 크거나 무한한 스트림에도 확장할 수 있습니다.

class Naturals:
    def __init__(self):
        self.n = 0
    def __iter__(self):
        return self
    def __next__(self):
        self.n += 1
        return self.n

it = iter(Naturals())
print(next(it), next(it), next(it))

반복자를 소비하는 내장 함수

많은 내장 함수는 어떤 반복자든 받을 수 있습니다. 예를 들어 sum, max, min, sorted, list, tuple이 있습니다. 이러한 함수는 StopIteration이 발생할 때까지 값을 가져옵니다.

class Count:
    def __init__(self, n):
        self.i = 0
        self.n = n
    def __iter__(self):
        return self
    def __next__(self):
        if self.i >= self.n:
            raise StopIteration
        self.i += 1
        return self.i

print(sum(Count(5)))
print(max(Count(5)))

빠른 확인

반복자 프로토콜에 대한 이해도를 확인해 보세요.

복습

반복자 프로토콜을 배웠습니다.

iter()는 반복자를 가져오고, next()는 값을 가져옵니다.
StopIteration은 끝을 알리며, for가 이를 처리합니다.
사용자 지정 반복자는 __iter__와 __next__를 구현합니다.
반복자는 한 번만 사용할 수 있고 메모리 효율적입니다.

다음에는 Python이 반복자를 대신 작성하도록 yield를 사용해 보겠습니다.

자주 묻는 질문

“반복자 프로토콜” 강의는 무료인가요?

네 — “반복자 프로토콜” 전체 내용을 이 웹사이트에서 무료로 읽을 수 있습니다. 인터랙티브하게 실습하려면(내장 코드 에디터와 24/7 AI 튜터), CoddyKit PRO로 업그레이드하면 Python Academy 강의 전체를 잠금 해제할 수 있습니다. Python Academy 강의에는 총 4개의 강의가 포함되어 있습니다.

“반복자 프로토콜”에서 뭘 배우나요?

__iter__와 __next__ 이해하기 브라우저에서 직접 실행하는 실습 코드로 Python Academy을(를) 배우며, 24/7 AI 튜터가 강의를 진행하면서 질문에 답변해줍니다.

Python Academy을(를) 시작하는 데 경험이 필요한가요?

사전 경험은 필요하지 않습니다. CoddyKit의 Python Academy은(는) 초급자부터 고급 학습자까지를 위해 구성되어 있으므로, 여기서 시작하거나 처음부터 시작할 수 있으며 자신의 속도대로 진행할 수 있습니다. 이것은 4개 중 1번째 강의입니다.

“반복자 프로토콜” 강의는 얼마나 걸리나요?

대부분의 CoddyKit 강의는 약 5~10분이 소요됩니다. 각 강의는 간결하고 인터랙티브하여 꾸준한 진행이 가능하며, 웹과 앱에서 중단한 부분부터 바로 시작할 수 있습니다.

이 Python Academy 강의에서 코드를 작성하고 실행할 수 있나요?

네. 모든 Python Academy 강의에는 내장 코드 에디터가 포함되어 있으므로, 브라우저에서 바로 실제 코드를 작성하고 실행한 후 즉시 AI 피드백을 받을 수 있습니다 — 로컬 설정이 필요 없습니다.

이 강의의 모든 강의

← Python Academy(으)로 돌아가기