[pandas] 데이터프레임 컬럼명 또는 행 인덱스 바꾸는 방법

안녕하세요. 비스카이비전입니다. 

 

오늘은 판다스 데이터프레임의 컬럼명과 행 인덱스를 바꾸는 방법에 대해 정리해보겠습니다.

 

 

위와 같은 데이터프레임이 있다고 가정하겠습니다. 세 개의 컬럼으로 구성되어 있는 데이터프레임입니다. 

 

데이터프레임 컬럼명 바꾸는 방법

우선 이 데이터프레임의 컬럼명을 1반 -> A반, 2반 -> B반, 3반 -> C반으로 변경해보겠습니다. 데이터프레임의 rename 메소드를 사용하시면 됩니다. 

 

import pandas as pd

dict_data = {'1반':['철수', '영희', '바둑'],
             '2반':['영수', '수정', '순자'],
             '3반':['영숙', '영철', '상철']}

df = pd.DataFrame(dict_data)
print(df)

df.rename(columns={'1반':'A반', '2반':'B반', '3반':'C반'}, inplace=True)
print(df)

 

rename 메소드 안 inplace 옵션이 True면, 원본 데이터프레임을 변경하겠다는 뜻입니다. 

 

 

컬럼명이 잘 바뀐 것을 확인하실 수 있습니다. 

 

데이터프레임 행 인덱스 바꾸는 방법

이번에는 0, 1, 2인 행 인덱스를 반장, 부반장, 총무로 바꿔보겠습니다. 역시 rename 메소드를 활용하면 됩니다. 이번에는 columns가 아니라 index를 수정해야 합니다.

 

import pandas as pd

dict_data = {'1반':['철수', '영희', '바둑'],
             '2반':['영수', '수정', '순자'],
             '3반':['영숙', '영철', '상철']}

df = pd.DataFrame(dict_data)
print(df)

df.rename(columns={'1반':'A반', '2반':'B반', '3반':'C반'}, inplace=True)
print(df)

df.rename(index={0:'반장', 1:'부반장', 2:'총무'}, inplace=True)
print(df)

 

 

행 인덱스가 반장, 부반장, 총무로 잘 바뀌었습니다.