[Python] 반복자 관련 메소드 - iterator

iter(x) -----------> x.__iter__() 호출

next(x) ----------< x.__next__() 호출

iter(x) 는 iterator 객체를 return 해야 한다. (결국 자신을 return 하면 된다.)

__next__ 함수에서는 더이상 넘길 값이 없으면 StopIteration exception을 발생시킨다

iterator 객체란 __iter__() 와 __next__() 를 구현한 class이다.

class Square(object): def __init__(self, start, end): self._number = start -1 self._end = end def __iter__(self): return self def __next__(self): if self._number == self._end: raise StopIteration else: self._number = self._number + 1 return self._number ** 2 if __name__ == '__main__': squares = iter(Square(1,5)) for item in squares: print(item, end=' ') print() squares = iter(Square(1,5)) print(next(squares)) print(next(squares)) print(next(squares)) print(next(squares))

1 4 9 16 25 1 4 9 16