파이썬 Pandas DataFrame 저장(to_csv)

DataFrame.to_csv(path_or_buf=None, sep=',', na_rep='', float_format=None, columns=None, header=True, index=True, index_label=None, mode='w', encoding=None, compression='infer', quoting=None, quotechar='"', line_terminator=None, chunksize=None, date_format=None, doublequote=True, escapechar=None, decimal='.', errors='strict', storage_options=None)

 

Pandas는 파이썬 데이터 분석 라이브러리로, 데이터를 다루는 데 유용한 기능을 제공합니다. 이 중 데이터프레임을 파일로 저장하는 to_csv() 함수를 이용하여, 데이터프레임을 CSV 파일로 저장하는 방법을 알아보겠습니다.

먼저, 예시로 사용할 데이터프레임을 생성해보겠습니다.

 

import pandas as pd 
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'], 'Age': [25, 30, 35, 40], 'Country': ['USA', 'Canada', 'UK', 'Australia']} 
df = pd.DataFrame(data)​

위 코드에서는 data라는 딕셔너리를 이용하여 DataFrame 객체인 df를 생성합니다. 이 데이터프레임은 Name, Age, Country 세 개의 열로 이루어져 있습니다.

이제 to_csv() 함수를 이용하여 df를 CSV 파일로 저장해보겠습니다.

df.to_csv('data.csv', index=False)

to_csv() 함수의 첫 번째 인자는 저장할 파일의 경로와 파일명을 지정합니다. 두 번째 인자인 index는 인덱스 값을 파일에 포함시킬지 여부를 지정합니다. index=False로 설정하면 인덱스 값을 파일에 포함시키지 않습니다.

만약 데이터프레임에서 특정 열만 선택하여 CSV 파일로 저장하고 싶다면, 다음과 같이 코드를 작성하면 됩니다.

df[['Name', 'Country']].to_csv('data.csv', index=False)

위 코드에서는 df에서 Name, Country 열만 선택하여 CSV 파일로 저장합니다.