[파이썬] for/else 사용하는 방법 (feat. while/else)

 

else 키워드를 for 문과 함께 사용할 수 있습니다.

for loop가 정상적으로 종료되면 else 절이 실행이 됩니다.

정상적인 종료라는 의미는 for 문이 break 구문을 만나지 않았다는 것을 의미 합니다.

for i in range(5):
    print(i)
else:
    print('completed for clause')

 

for 문을 사용하다보면. for 문이 break 문을 통해서 강제 종료 되었는지 아닌지 확인할 필요가 있습니다.

소수를 구하는 예제 프로그램을 통해서 설명 하겠습니다.

반복문 내에서 % 연산자를 통해서 소수를 체크하는데 중간에 break 문을 만났다는 것은 소수가 아닌 것입니다.

그래서, prime_num 플래그 변수를 하나 둬서 prime_num을 False로 체크하고 break 문을 통해서 반복문을 종료합니다.

prime_num인 False이기 때문에 소수가 아니므로 prime_nums 리스트에 추가하지 않습니다.

# 소수 구하기 예제

prime_nums = []

for n in range(2, 20):
    prime_num = True
    for x in range(2, n):
        if n % x == 0:
            prime_num = False
            break
            
    if prime_num:
        prime_nums.append(n)
        
print(prime_nums)

위 예제 코드에 for/else 구문을 적용해 보겠습니다.

굉장히 코드가 심플해졌습니다. prime_num 플래그 변수도 사라졌습니다. 반복문을 완벽히 돌았다는 것은 prime number를 의미하므로 else 구문에서 prime_nums 리스트에 추가하면 됩니다.

# 소수 구하기 예제 

prime_nums = []

for n in range(2, 20):
    for x in range(2, n):
        if n % x == 0:
            break
    else:
        prime_nums.append(n)
        
print(prime_nums)

 

while 키워드에도 else가 가능합니다.

 

 

참고 자료

https://book.pythontips.com/en/latest/for_-_else.html

https://pyneng.readthedocs.io/en/latest/book/06_control_structures/for_while_else.html