<Python> Iterator, Generator

1. Iterator

__iter__, __next__를 가진 객체를 iterator protocol을 지원한다고 일컫는다.

class Counter:
    def __init__(self, limit):
        self.num = 0
        self.limit = limit

    def __iter__(self):
        return self

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

for i in Counter(10):
    print(i, end=' ')

 

클래스에서 __getitem__ 메서드만 구현해도 이터레이터가 된다.

class Counter:
    def __init__(self, limit):
        self.num = 0
        self.limit = limit

    def __getitem__(self, index):
        if index < self.limit:
            return index * 3
        else:
            raise IndexError

for i in Counter(10):
    print(i, end=' ')

 

2. Generator

Generator는 Iterator와 유사하나 yield를 사용하여 구현한다.

def generator(count):
    list = [1, 2, 3, 4, 5]
    t = 0
    while t < count:
        yield list[t % len(list)]
        t += 1

g = generator(53)
for i in g:
    print(i, end=' ')

yield from은 반복가능한 객체 / 이터레이터 / 제너레이터 객체 등에 사용 가능하다.

def generator():
    list = [1, 2, 3, 4, 5]
    yield from list

g = generator()
for i in g:
    print(i, end=' ')