초보자를 위한 파이썬 300제 // 126 if문

 

문제

우편번호는 5자리로 구성되는데, 앞의 세자리는 구를 나타낸다. 예를들어, 강북구의 경우 010, 011, 012 세 자리로 시작한다.

-	0	1	2	3	4	5	6	7	8	9
01	강북구	강북구	강북구	도봉구	도봉구	도봉구	노원구	노원구	노원구	노원구

사용자로 부터 5자리 우편번호를 입력받고 구를 판별하라.

>> 우편번호: 01400
도봉구

내풀이

i = input("우편번호: ")[:3]
if i in ["010", "011", "012"]:
    print("강북구")
elif i in ["014", "015", "016"]:
    print("도봉구")
else:
    print("노원구")
    
결과값
우편번호: 01400
도봉구

정답

우편번호 = input("우편번호: ")
우편번호 = 우편번호[:3]
if 우편번호 in ["010", "011", "012"]:
    print("강북구")
elif 우편번호 in ["014", "015", "016"]:
    print("도봉구")
else:
    print("노원구")

 

슬라이싱을 이용해 원하는 값만 비교를 하였다.