문제 풀이/Programmers

[python3][level 2] 2016년

[잉간] 2018. 4. 16. 14:20


예전에 비슷한 문제를 풀었을 때는 일일히 계산했다.

이번에는 편한 방법이 없을까 찾아보았더니 입력받은 날짜를 요일로 반환해주는 모듈이 있길래 사용해보았다.

문제풀이

import datetime

def getDayName(a,b):
    day = ['MON','TUE','WED','THU','FRI','SAT','SUN']
    return day[datetime.date(2016, a, b).weekday()]

#아래 코드는 테스트를 위한 출력 코드입니다.
print(getDayName(5,24))