pandas를 사용하여 데이터프레임(DataFrame)을 엑셀(Excel) 파일로 저장하는 방법

들어가며

이전 포스팅에 이어 이번에는 DataFrame 객체를 엑셀 파일로 저장하는 방법에 대해서 정리해보았다. 예시에서 사용할 DataFrame 객체의 데이터는 다음과 같다.

columns = ["번호", "케이블", "레이스웨이"]
data = [
    [1, "cable01", "raceway01"],
    [2, "cable01", "raceway02"],
    [3, "cable01", "raceway03"],
    [4, "cable01", "raceway04"],
    [5, "cable01", "raceway05"],
    [6, "cable02", "raceway03"],
    [7, "cable02", "raceway05"],
    [8, "cable02", "raceway07"],
    [9, "cable03", "raceway10"],
    [10, "cable03", "raceway11"],
    [11, "cable03", "raceway12"]
]

엑셀 파일로 저장하기

DataFrame 객체를 엑셀 파일로 저장하기 위한 코드는 다음과 같다.

import os
import pandas as pd

excel = "C://Users/CHOEWY/Desktop/new_exam.xlsx"
df = pd.DataFrame(data, columns)
df.to_excel(excel, sheet_name="New", index=False)

os.startfile(excel)

먼저, 새롭게 저장할 엑셀 파일의 경로를 excel 변수에 지정하고, 위의 데이터를 DataFrame 객체로 변환한다. 이어서 to_excel 함수를 통해 해당 객체를 엑셀 파일로 저장하였는데, DataFrame 자체에 설정된 인덱스를 사용하지 않기 위하여 index=False로 설정하였다. 이어서 os 모듈의 startfile 함수를 통해 저장된 엑셀 파일을 바로 실행해주었다.

마치며

다음에는 pandas를 사용하여 두 개의 엑셀 파일의 데이터를 가공 후 결과물을 생성하는 방법에 대해서 다뤄보겠다.