PROGRAMMERS Python Lv1. 2016년

최대 1 분 소요

이번 문제는 2016년이다.

이 문제는 월과 날짜를 입력받아 그 날이 무슨 요일인지

출력하는 문제이다.

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

간단하게 datetime을 사용하면 된다.

우선 요일을 지정하는 리스트를 만들어주고,

datetime.datetime에 년, 월, 일 순으로 넣어준다.

그러면 weekday를 통해 요일이 반환된다.

weekday는 0: 월요일부터 6:일요일 순서이다.

혹시 제 풀이에 오류가 있거나

더 좋은 방법이 있다면 댓글 남겨주시면 감사드립니다.

댓글남기기