Pandas: dataframe의 특정 열을 csv 파일로 내보내기

다음 구문을 사용하여 Pandas DataFrame의 특정 열만 CSV 파일로 내보낼 수 있습니다.

 df. to_csv (' my_data.csv ', columns=[' col1 ',' col4 ',' col6 '])

columns 인수는 DataFrame에서 CSV 파일로 내보낼 특정 열을 팬더에게 알려줍니다.

다음 예에서는 실제로 이 구문을 사용하는 방법을 보여줍니다.

예: Pandas DataFrame의 특정 열을 CSV 파일로 내보내기

다양한 농구 선수에 대한 정보가 포함된 다음과 같은 pandas DataFrame이 있다고 가정합니다.

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 18 5 11
1 B 22 7 8
2 C 19 7 10
3 D 14 9 6
4 E 14 12 6
5 F 11 9 5
6 G 20 9 9
7:28 4 12

to_csv() 함수를 사용하여 DataFrame을 CSV 파일로 내보내면 pandas는 기본적으로 모든 열을 내보냅니다.

 #export DataFrame to CSV file
df. to_csv (' basketball_data.csv ')

CSV 파일은 다음과 같습니다.

DataFrame의 모든 열은 CSV 파일에 포함됩니다.

DataFrame의 특정 열만 CSV 파일로 내보내려면 columns 인수를 사용할 수 있습니다.

예를 들어 다음 구문을 사용하여 팀 및 리바운드 열만 CSV 파일로 내보낼 수 있습니다.

 #export only team and rebounds columns from DataFrame to CSV file
df. to_csv (' basketball_data.csv ', columns=[' team ',' rebounds '] )

CSV 파일은 다음과 같습니다.

팀 및 리바운드 열만 CSV 파일에 포함됩니다.

참고 : pandas to_csv() 함수에 대한 전체 문서는 여기에서 찾을 수 있습니다.

추가 리소스

다음 튜토리얼에서는 Pandas에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

Pandas에서 헤더 없는 CSV 파일로 데이터를 내보내는 방법
Pandas에서 여러 CSV 파일을 병합하는 방법
Pandas의 CSV 파일에서 특정 줄만 읽는 방법