LANGUAGE/python

[Python] for-else 문

뚜비두빱 2022. 6. 12. 14:58

가끔 파이썬 코딩을 할 때 다음과 같은 경우가 있다.

answer = []
for i in arr:
    if i % 3 == 0:
        answer.append(i)
        flag = True
        break
if not flag:
    answer.append(0)

원래는 위와 같이 flag를 쓰거나 함수를 만들어서 값을 return 해주는 방법으로 바꿔줬는데

더 좋은 방법이 있다. 파이썬에는 다음과 같이 for-else문이 가능하다.

answer = []
for i in arr:
    if i % 3 == 0:
        answer.append(i)
        break
else:
    answer.append(0)