pandas 데이터 분석 - 칼럼 이름 변경하기(rename, df.column =[], lambda)

데이터를 분석하다 보면 칼럼 이름을 바꾸거나 순서를 바꿔야 하는 경우가 왕왕 있습니다.

오늘은 pandas 데이터 프레임에서 칼럼이름을 변경하는 법에 대해서 알아보겠습니다.

예시 데이터 프레임

#예제용 데이터프레임
df = pd.DataFrame([[1,2,3],[0,2,4],[0,3,3],[1,2,3]],columns=['c1','c2','c3'])

예시 데이터 프레임

 

 

1. rename

먼저 rename 함수를 이용해서 바꿔보겠습니다.

rename 함수는 지정되는 컬럼을 지정한 이름으로 바꿔주는 함수입니다.

소수의 칼럼의 이름을 바꿀 때, 도움이 되는 함수입니다.

 

- 하나의 칼럼이름을 바꿀 때

c1이라는 칼럼을 d1으로 바꿔보겠습니다.

df.rename(columns = {'c1':'d1'},inplace=True)

rename - 결과1

 

- 여러 개의 칼럼이름을 바꿀 때

칼럼 중 c2, c3의 이름을 d2, d3로 바꿔보겠습니다.

df.rename(columns = {'c2':'d2','c3':'d3'},inplace=True)

 

rename - 결과2

 

2. df.columns = []

rename은 특정 칼럼의 이름을 바꿔주는 함수라면 columns 메서드는 전체를 한방에 바꿔주는 메서드라고 생각하시면 됩니다.

많은 숫자의 칼럼명을 한 번에 바꿀 때, 효율적인 방법입니다.
 

주의사항

- df.columns를 활용할 때는 칼럼 개수는 정확하게 일치시켜줘야 합니다.

- df.columns를 순서를 잘 생각해서 리스트에 넣어줘야 합니다.

 

- d1, d2, d3 칼럼명을 한방에 c1, c2, c3로 바꿔줘 보겠습니다.

df.columns = ['c1','c2','c3']

 

df.columns = [] - 결과

 

3. 칼럼명에 접두사 접미사 붙여보기

- lambda 함수를 이용해서 dataframe 칼럼 앞에 'data_'라는 접두사를 붙여보겠습니다.

df.rename(columns=lambda x : "data_" + x, inplace = True)

 

lambda - 결과1

 

- 이번에는 '_data'라는 접미사를 붙여보겠습니다.

df.rename(columns=lambda x : x + "_data", inplace = True)

lambda - 결과 2