[python] 파이썬 workalendar를 사용하여 특정 국가의 휴일 목록을 CSV 파일로 저장하기
휴일 관련 데이터를 사용할 때, workalendar는 파이썬 라이브러리로서 유용한 도구입니다. 이 라이브러리를 사용하여 특정 국가의 휴일 목록을 가져와서 CSV 파일로 저장하는 방법을 알아보겠습니다.
작업 환경 설정
- 파이썬의
workalendar라이브러리를 설치하려면 다음 명령을 실행합니다.
pip install workalendar
- 적절한 편집기를 사용하여 새로운
.py확장자로 파일을 생성합니다.
코드 작성
다음은 예제 코드입니다.
from workalendar.registry import registry
# 특정 국가의 휴일 목록을 가져오기 위해 국가 코드를 설정합니다. (ex: 'KR'는 대한민국)
country_code = 'KR'
# 특정 국가의 휴일 관련 클래스 가져오기
# 휴일 정보는 Gregorian, Lunar, Islamic 등 다양한 달력 시스템에서 제공할 수 있습니다.
# 이 예제에서는 Gregorian 달력을 사용합니다.
cal_class = registry.get_calendar_class(country_code)
# 특정 년도의 휴일 목록 가져오기
cal = cal_class()
# 휴일 목록을 CSV 파일로 저장합니다.
with open('holidays.csv', 'w') as file:
file.write("Date, Holiday\n") # CSV 파일의 헤더를 작성합니다.
for holiday in cal.holidays():
date = holiday[0]
name = holiday[1]
file.write(f"{date}, {name}\n") # 휴일 날짜와 이름을 CSV 파일에 작성합니다.
실행 결과
위의 코드를 실행하면 holidays.csv라는 파일이 생성되고, 해당 국가의 휴일 목록이 CSV 형식으로 저장됩니다.