[파이썬] 문제 : 감염병 백신 주사 요일 계산

n = input('출생연도를 입력하시오: ')
w = ''
if n[-1] == '0' or n[-1] == '5':
    w = '월'
elif n[-1] == '1' or n[-1] == '6':
    w = '화'
elif n[-1] == '2' or n[-1] == '7':
    w = '수'
elif n[-1] == '3' or n[-1] == '8':
    w = '목'
elif n[-1] == '4' or n[-1] == '9':
    w = '금'

print(f"{w}요일에 접종가능합니다.")

 

다음과 같이 딕셔너리를 사용할 수 있습니다.

w_dic = {'0':'월','1':'화','2':'수','3':'목','4':'금','5':'월','6':'화','7':'수','8':'목','9':'금'}
n = input('출생연도를 입력하시오: ')
print(f"{w_dic[n[-1]]}요일에 접종가능합니다.")   # 딕셔너리에서 key로 해서 찾음. [-1]은 리스트의 마지막 요소

 

다음과 같이 리스트를 사용할 수 있습니다.

w_list = ['월','화','수','목','금']
n = input('출생연도를 입력하시오: ')
w = ''
if int(n[-1]) <= 4:               # 4 까지는 리스트의 인덱스로 사용하면 됨
    w = w_list[int(n[-1])]
else:                             # 4 이상이면
    w = w_list[int(n[-1])-5]      # -5를 하면 0,1,2,3,4와 같이 되어 리스트의 인덱스로 사용 가능

print(f"{w}요일에 접종가능합니다.")