Pandas: csv 파일을 가져올 때 열 이름 설정

CSV 파일을 Pandas로 가져올 때 다음 기본 구문을 사용하여 DataFrame의 열 이름을 정의할 수 있습니다.

 colnames = [' col1 ', ' col2 ', ' col3 ']

df = pd. read_csv (' my_data.csv ', names=colnames)

names 인수는 DataFrame의 열에 사용하려는 이름 목록을 가져옵니다.

이 인수를 사용하면 팬더에게 CSV 파일의 첫 번째 줄을 헤더 줄로 사용하는 대신 DataFrame의 첫 번째 줄로 사용하도록 지시하는 것입니다.

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

예: CSV 파일을 Pandas로 가져올 때 열 이름 설정

Players_data.csv 라는 다음과 같은 CSV 파일이 있다고 가정해 보겠습니다.

파일에서 첫 번째 행에 열 이름이 포함되어 있지 않음을 알 수 있습니다.

read_csv() 함수를 사용하여 CSV 파일을 가져오면 pandas는 첫 번째 행의 값을 DataFrame의 열 이름으로 사용하려고 시도합니다.

 import pandas as pd

#import CSV file
df = pd. read_csv (' players_data.csv ')

#view resulting DataFrame
print (df)

   At 22 10
0 B 14 9
1 C 29 6
2 D 30 2
3 E 22 9
4 F 31 10

그러나 CSV 파일을 가져올 때 names 인수를 사용하여 자체 열 이름을 지정할 수 있습니다.

 import pandas as pd

#specify column names
colnames = [' team ', ' points ', ' rebounds ']

#import CSV file and use specified column names
df = pd. read_csv (' players_data.csv ', names = colnames)

#view resulting DataFrame
print (df)

  team points rebounds
0 to 22 10
1 B 14 9
2 C 29 6
3 D 30 2
4 E 22 9
5 F 31 10

CSV 파일의 첫 번째 줄은 더 이상 헤더 줄로 사용되지 않습니다.

대신, names 인수를 사용하여 지정한 열 이름이 이제 열 이름으로 사용됩니다.

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

추가 리소스

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

Pandas: CSV 파일을 읽을 때 줄을 건너뛰는 방법
Pandas: 기존 CSV 파일에 데이터를 추가하는 방법
Pandas: usecols 인수와 함께 read_csv를 사용하는 방법