본문 바로가기
Python/Pandas

pandas df.to_csv() 함수 활용하기

by PySun 2023. 11. 16.
반응형

df.to_csv() 함수는 Pandas 데이터프레임을 CSV 파일로 저장하는 데 사용됩니다. 이 함수를 사용하면 데이터프레임을 표 형식의 CSV 파일로 내보낼 수 있습니다.

메서드 구문:

df.to_csv(
    path_or_buf,
    sep=',',
    na_rep='',
    index=True,
    header=True,
    columns=None,
    mode='w',
    encoding=None
)

주요 매개변수:

  • path_or_buf: CSV 파일의 경로 또는 파일 객체.
  • sep: 열을 구분하는 문자 또는 문자열. 기본값은 쉼표(,)입니다.
  • na_rep: 결측값 (NaN)를 표현하는 문자열.
  • index: 인덱스를 CSV 파일에 포함할지 여부를 설정. 기본값은 True입니다.
  • header: 열 이름을 포함할지 여부를 설정. 기본값은 True입니다.
  • columns: 저장할 열의 이름을 나타내는 리스트를 지정. 기본값은 None으로 모든 열을 저장합니다.
  • mode: 파일 쓰기 모드. 'w' (기본값) 또는 'a'를 사용할 수 있습니다.
  • encoding: 파일의 문자 인코딩을 지정. 기본값은 시스템 기본 인코딩입니다.

예시 코드:

import pandas as pd

# 샘플 데이터프레임 생성
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'City': ['New York', 'Los Angeles', 'Chicago']
}
df = pd.DataFrame(data)

# 데이터프레임을 CSV 파일로 저장
df.to_csv('sample.csv', sep='|', index=False)

# 저장된 CSV 파일을 다시 불러와서 확인
loaded_df = pd.read_csv('sample.csv', sep='|')
print(loaded_df)

이 코드에서는 먼저 샘플 데이터프레임을 생성하고, 그 후 df.to_csv() 메서드를 사용하여 데이터프레임을 'sample.csv'라는 파일로 저장합니다. 이 파일은 파이프 ('|')로 구분되며, 인덱스는 저장되지 않도록 설정했습니다. 그런 다음 pd.read_csv() 함수를 사용하여 저장된 CSV 파일을 다시 불러와서 데이터를 확인합니다.
df.to_csv() 함수를 사용하면 데이터프레임을 다양한 형식의 CSV 파일로 저장할 수 있으며, 필요에 따라 설정을 조절하여 저장할 수 있습니다.

반응형