안녕하세요! 오늘은 Python의 pandas 라이브러리를 사용하여 DataFrame을 엑셀 파일에 저장할 때, 여러 시트를 만드는 방법에 대해 소개하고자 합니다. 여러 개의 DataFrame을 하나의 엑셀파일로 만들고 싶을 때 자주 사용하는 방법인데요. 필요할 때마다 찾아 쓰는 코드이다 보니 공유하면 좋을 것 같아 가져와봤습니다.
pandas 라이브러리에서 DataFrame을 엑셀 파일로 저장할 때 사용하는 메서드는 to_excel()를 사용하지만, 여러 개의 시트를 만들려면 ExcelWriter 객체를 사용해야 합니다. ExcelWriter 객체를 생성하고, to_excel() 메서드를 사용해 여러 개의 DataFrame을 각각 다른 시트에 저장할 수 있습니다. 아래 코드는 두 개의 DataFrame을 엑셀 파일의 서로 다른 시트에 저장하는 예시 코드입니다. 코드를 실행하면 'DF_1', 'DF_2' 시트를 가진 'output.xlsx' 엑셀 파일을 생성할 수 있습니다.
import pandas as pd
# 예제 DataFrame 생성
df_1 = pd.DataFrame({
'Column1': [1, 2, 3],
'Column2': ['A', 'B', 'C']
})
df_2 = pd.DataFrame({
'Column1': [4, 5, 6],
'Column2': ['D', 'E', 'F']
})
# 엑셀 파일 경로 설정
xlxs_dir = 'output.xlsx'
# ExcelWriter 객체를 사용하여 DataFrame을 엑셀 파일에 쓰기
with pd.ExcelWriter(xlxs_dir) as writer:
df_1.to_excel(writer, sheet_name='DF_1')
df_2.to_excel(writer, sheet_name='DF_2')
데이터 분석을 할 때 엑셀파일을 저장하는 상황에서 여러 개의 시트로 한번에 저장하고 싶을 때 유용한 코드입니다. 도움이 되었길 바라며, 오늘도 읽어주셔서 감사합니다.