[Python][프로그래머스][Level1] 2016년

[Python][프로그래머스][Level1] 2016년

코드

from datetime import datetime
def solution(a, b):
    answer=['MON','TUE','WED','THU','FRI','SAT','SUN']
    return answer[datetime(2016,a,b).weekday()]

datetime 라이브러리를 쓰면 쉽게 풀 수 있다.

datetime.weekday() 는 월화수목금토일 순으로 0~6을 반환한다.

Discussion and feedback