[파이썬] 데이터 프레임(dataframe), csv 또는 엑셀 파일로 저장하기

데이터프레임을 통해 데이터를 다룬 뒤, 이를 저장하기 위해선 csv 파일 또는 엑셀(excel, xls) 파일로 저장해야 하는데요, 'to_csv', 'to_excel' 이라는 모듈을 이용하여 데이터프레임을 각 csv 파일, 엑셀 파일로 저장할 수 있습니다.

 

 

1. to_csv : csv 파일로 저장하기

데이터프레임을 csv 파일로 저장해주는 'to_csv' 는 데이터프레임.to_csv("저장할 경로+파일명.csv") 와 같이 입력하면 되는데요, 만약 'subject' 라는 이름의 데이터프레임이 있을 때, 이 데이터프레임을 test 폴더에 "dataframe1" 이라는 이름의 csv 파일로 저장하려면 다음과 같이 코드를 입력하면 됩니다.

 

subject.to_csv("c:\\test\\dataframe1.csv", encoding="utf-8-sig", index=False)

 

한편, 위 코드를 보면 뒤에 'encoding="utf-8-sig", index=False' 라고 입력된 것을 볼 수 있는데요, 이는 encoding 형식과 index 형식을 지정해 준 것으로, encoding 부분은 이를 생략하면 에러가 날 수 있습니다.

 

 

2. to_excel : excel 파일로 저장하기

 반면, dataframe을 excel 파일로 저장해주는 모듈은 'to_excel'로 to_csv와 동일하게 사용하면 됩니다.

따라서 만약 마찬가지로 'subject'라는 데이터프레임을 'test' 폴더에 'dataframe1' 이라는 엑셀파일로 저장하고 싶다면 다음과 같이 입력하면 됩니다.

 

subject.to_excel("c:\\test\\dataframe1.xls",  index=False) 

 

엑셀파일로 저장할 때는 encoding 부분을 따로 설정해주지 않아도 오류가 나지 않는답니다.