[python] 조건문에서의 이터레이터 객체 사용하기

파이썬에서 이터레이터(iterator) 객체를 조건문에서 활용하여 효과적인 코드를 작성할 수 있습니다. 이터레이터 객체를 사용하면 데이터를 한 번에 처리하는 대신 필요할 때마다 값을 생성하여 메모리를 절약할 수 있습니다. 이터레이터 객체를 조건문에서 활용하는 방법에 대해 알아보겠습니다.

이터레이터 객체란?

이터레이터(iterator)는 __iter__()__next__() 메서드를 구현하는 객체입니다. __iter__() 메서드는 이터레이터 객체 자신을 반환하고, __next__() 메서드는 데이터를 반환하며 다음 값을 가리킵니다. 이터레이터 객체는 for문이나 next() 함수와 함께 사용할 수 있습니다.

조건문에서 이터레이터 객체 사용하기

이터레이터 객체를 조건문에서 사용할 때 next() 함수를 호출하여 다음 값이 있는지 확인하고, 값이 있을 경우에만 해당 값을 사용합니다.

# 이터레이터 객체 생성
my_iter = iter([1, 2, 3])

# 이터레이터 객체 사용
try:
    while True:
        value = next(my_iter)
        if value % 2 == 0:
            print(f'짝수: {value}')
        else:
            print(f'홀수: {value}')
except StopIteration:
    pass

위 예제에서는 iter() 함수를 사용하여 리스트를 이터레이터 객체로 만든 후, try-except 구문을 통해 StopIteration 예외를 처리하며 이터레이터 객체를 조건문에서 사용하였습니다.

결론

이번 글에서는 파이썬의 이터레이터 객체를 조건문에서 활용하는 방법에 대해 알아보았습니다. 이터레이터 객체를 조건문에서 활용하면 데이터를 효율적으로 처리할 수 있으며, 메모리를 절약할 수 있습니다. 이를 통해 보다 효율적이고 가독성 있는 코드를 작성할 수 있습니다.

참고 자료