파이썬의 iterator 사용법

1. 개요

파이썬으로 알고리즘을 풀다보면 간혹가다 iterator를 써야하는 경우가 생긴다. 이번 기회에 한번 사용법을 알아보자

 

 

2. 사용법

iterable한 객체에서 반복자를 얻어오는 방법은 다음과 같다

iterable = [ 1, 2, 3, 4, 5 ]
it = iter(iterable)

 

값을 꺼내오려면 다음과같이 next 함수를 사용하면 된다

iterable = [ 1, 2, 3, 4, 5 ]
it = iter(iterable)
next(it) #1
next(it) #2
next(it) #3
next(it) #4
next(it) #5

 

 

3. hasnext

파이썬의 반복자에는 크나큰 결점이 있다. hasnext 메소드가 없다. 더이상 next가 없는 상황에서 next를 호출하면 exception을 일으킨다.

이를 해결하기 위해서 아래처럼 next 함수에 default값을 주어 예외처리를 해주자

 

 

iterable = [ 1, 2, 3, 4, 5 ]
it = iter(iterable)

while True:
	val = next(it, None)
    if val == None:
    	break
    print(val)