[Python pandas] pandas rename 함수 : DataFrame의 컬럼명을 변경해보자

오늘은 pandas의 rename 함수에 대해서 알아보겠습니다.

먼저, pandas.DataFrame을 통해 df라는 dataframe을 작성해보겠습니다.

import pandas as pd
df = pd.DataFrame({
    'x1' : [1,2,3,4],
    'x2' : [5,6,7,8],
    'y1' : [1,3,5,7],
    'y2' : [2,4,6,8],
})

 

dataframe이 어떻게 생겼는지 확인해보면 아래와 같습니다.

df

 

이 df라는 dataframe에서 컬럼명인 x1과 x2를 각각 a1, a2로 바꾸고싶다면, 아래와 같이 간단하게 바꿀 수 있습니다.

df.columns = ['a1','a2','y1','y2']

 

하지만, x1 하나만 바꾸고싶다면, 아래와 같이 다른 컬럼명들도 일일이 다 입력해줘야하는 번거로움이 발생합니다.

df.columns = ['a1','x2','y1','y2']

 

이러한 과정을 좀 더 단순하게 만들어주는 함수가 뭘까요?

바로! pandas의 rename 함수로, 아래와 같이 해당 컬럼만 변경하도록 입력해주면됩니다.

df.rename(columns={'x1':'a1'}, inplace=True)

 

코드가 살짝 길어지는 것 같지만, dataframe의 컬럼이 수십개, 수백개인 경우에는

필요한 컬럼만 명시해주면되는 rename 함수가 훨씬 간결해집니다.

또한, pandas의 rename 함수는 inplace를 지원해주기때문에 아래와 같이 df에 덮어씌우기를 하지않아도 됩니다.

df = df.rename(columns={'x1':'a1'})

 

pandas의 rename 함수에 대한 상세한 내용은 pandas의 공식홈페이지에서 확인하세요!

pandas.DataFrame.rename — pandas 2.1.0 documentation