[Code Solution] 데이터프레임 column/index 이름 변경하기

데이터프레임 column/index 이름 변경하기

데이터프레임 형태로 제공되는 데이터를 사용하다보면, column 또는 index에 설정되어 있는 이름이 마음에 들지 않거나, 보다 파악하기 쉬운 이름으로 변경하고싶은 경우가 있습니다.

이러한 경우에는 pandas 라이브러리의 데이터프레임에 기본적으로 내장되어 있는 rename() 함수를 사용하면 손쉽게 변경이 가능합니다.

rename() 함수를 사용하여 이름을 변경할 때, column 이름을 변경하고 싶다면 내부에 columns={'바꾸고 싶은 이름1' : '새로운 이름1', '바꾸고 싶은 이름2' : '새로운 이름2'} 의 형태를 갖는 파라미터를 입력해주면 되고, index 이름을 변경하고 싶다면 내부에 index={'바꾸고 싶은 이름1' : '새로운 이름1', '바꾸고 싶은 이름2' : '새로운 이름2'}의 형태를 갖는 파라미터를 입력해주어야 합니다.

추가적으로 rename() 함수 내부에는 inplace라는 파라미터가 존재하는데 만약 inplace를 False로 설정해주면 원본 객체를 수정하는 것이 아닌 새로운 데이터프레임 객체를 반환하고 inplace를 True로 설정해주면 원본 객체를 수정하는 방향으로 함수가 작동됩니다.

import pandas as pd

# 데이터프레임 생성
data = pd.Dataframe([[30, 'samsung'], [32, 'lg']], columns=['연령','회사'], index=['사람1', '사람2'])

# Column 이름 변경
data.rename(columns={'연령':'나이', '회사':'소속'}, inplace=True)

# Index 이름 변경
data.rename(index={'사람1':'철수', '사람2':'영희'}, inplace=True)

python version: 3.7.13