파이썬 | pandas | 데이터프레임의 전체 열(Column) 이름 가져오기

이 글은 데이터프레임에서 전체 열(Column) 이름을 가져오는 방법에 대해 정리한 글이다.

 

pandas 라이브러리를 이용하여 데이터를 불러오는 방법은 아래 글을 참고하면 된다.

파이썬 | pandas | csv, 텍스트(txt), 엑셀(xlsx) 데이터 불러오기

파이썬 | pandas | csv, 텍스트(txt), 엑셀(xlsx) 데이터 불러오기

---

칼럼 명 가져오기

가상의 성적 데이터를 활용하여, 이 데이터에 있는 칼럼 명을 모두 가져오는 방법에 대해 알아보자.

100명의 학생의 성적 데이터

위 데이터는 Name부터 Chemistry까지 총 8개의 칼럼으로 구성되어 있는 가상의 데이터이다.

 

dataframe.columns를 이용하면 데이터프레임의 칼럼 명을 가져올 수 있다.

import pandas as pd

# 데이터 불러오기
df = pd.read_csv('./sample_data/grade_sample_data.csv', encoding='cp949')

# 칼럼 명 추출
cols = list(df.columns)

cols를 출력해보면 칼럼 명을 리스트 형태로 모두 가져온 것을 볼 수 있다.

 

리스트를 활용하여 원하는 칼럼만 추출하거나 여러 개의 파일을 컨트롤 할 때 유용하게 사용할 수 있다.