[Pandas] 데이터 프레임 컬럼명 확인하는 방법

데이터프레임의 컬럼명 확인하는 방법과 컬럼명을 변경하는 방법을 알아보겠습니다.

 

1. 데이터 프레임 컬럼명 확인하기

우선은 예제로 사용할 데이터 프레임을 생성해보겠습니다.

import pandas as pd

df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
})python

 

생성한 데이터 프레임의 컬럼을 가져오기 위해서는 데이터프레임 이름에 columns로 접근 합니다.

print(df.columns)python

columns를 사용하면 Index로 컬럼값을 가져옵니다. 이를 리스트 형태로 바꾸기 위해서는 to_list() 함수를 이용합니다.

column_names = df.columns.tolist()
print(column_names)python

 

반응형

 

2. 컬럼명 변경하기

2.1. 전체 컬럼명 변경하기

 df.columns=['col1', 'col2','col3']
 print(df.columns)python

변경 할 컬럼명을 리스트에 넣고 columns에 할당하면 전체 컬럼명을 변경할 수 있습니다. 이 때 데이터 프레임의 컬럼수와 리스트 항목수가 일치해야만 컬럼명이 정상적으로 변경됩니다.

 

2.2. 일부 컬럼명 변경하기

일부 컬럼명을 변경할 떄는 rename() 함수를 이용합니다.

df.rename(columns={'A': 'X'}, inplace=True)
print(df)python

A 컬럼의 이름이 X로 변경되었습니다.

 

이상 데이터프레임의 컬럼명을 확인하는 방법과 변경하는 방법을 알아보았습니다.