Pandas DataFrame를 csv로 저장하고 로드하기

Pandas DataFrame에 저장된 데이터셋을 파일로 저장하고 로드하면 좋겠다는 생각이 들어서 여러모로 검색을

하여서 조금씩 정보를 찾았다.

일단 csv로 저장하는 방법이다. 매우 간단하다.

import pandas as pd

time_pd = pd.DataFrame(0., columns=col, index=time_range)

time_pd.to_csv("filename.csv", mode='w')

일단 위의 소스는 time_pd라는 임의의 DataFrame을 생성한 뒤 바로  csv파일로 저장하는 소스코드이다.

mode에 인자값을 ‘a’ 로 해주면 덮어쓰기가 아닌 추가로 내용을 쓸 수 있다.

time_pd2.to_csv("filename.csv", mode='a', header=False)

이렇게 하면 위에서 말한대로 추가가 되고 header를 False 값을 주면 추가되는 값에 header가 찍히지 않는다.

 

dataset = pd.read_csv("filename.csv", index_col=0)

print(dataset)

반대로 csv파일을 불러오는 소스코드이다. 인덱스 column을 0번째 줄로 정하겠다는 의미이다.

참으로 유용하면서 간단한 모듈이라고 할 수 있겠다.