Overview

iterable은 반복 가능한 객체인 반면, iterator는 반복을 수행하는 객체이다. 새로운 데이터 타입을 만들 때, 그 데이터 타입이 반복 가능한 성질을 갖도록 할 때 이터레이터를 사용한다.

Iterator

Iterator(이터레이터)는 순서대로 하나씩 값을 리턴할 수 있는 객체이다.

• iter()함수를 통해 iterable로부터 iterator를 생성
• next()함수 또는 __next__()메서드를 통해 값을 리턴
• 내부적으로 __next__()와 __iter__()메서드를 가짐
• 끝에 도달하면 StopIteration 예외 발생

nums = [10, 20, 30]
it = iter(nums)

print(next(it))
print(next(it))
print(next(it))
print(next(it))  # E

nums = [10, 20, 30]
it = iter(nums)

print(it.__next__())
print(it.__next__())
print(it.__next__())
print(it.__next__())  # E

• 결과

[Option] Class로 직접 구현

class MyIterator:
    def __init__(self, x):
        self.x = x
        self.position = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.position >= len(self.x):
            raise StopIteration
        result = self.x[self.position]
        self.position += 1
        return result

i = MyIterator(["apple", "banana", "cherry"])
for item in i:
    print(item)

class MyIterator:
    def __init__(self, x):
        self.x = x
        self.position = len(self.x) -1

    def __iter__(self):
        return self

    def __next__(self):
        if self.position < 0:
            raise StopIteration
        result = self.x[self.position]
        self.position -= 1
        return result

i = MyIterator(["apple", "banana", "cherry"])
for item in i:
    print(item)