DataFrame 특정 컬럼명 수정, 전체 컬럼명 수정하는 방법

 

Pandas DataFrame의 특정 컬럼명 수정이나 전체 컬럼명을 수정할 때, 저는 아래 2가지 방법을 이용합니다.

 

# df의 컬럼 : col1, col2, col3

# 방법1
df.rename(columns = {'col2':'변경할컬럼명'}, inplace = True)
 
# 방법2
df.columns = ['col1', '변경할컬럼명', 'col3']

 

제가 실무를 하면서 다음과 같은 경우도 존재했었는데요,

(저만 있었을 수도...)

 

#1) 컬럼명 전체를 대/소문자로 변경

# 방법
df.columns = [x.upper() for x in df.columns] # 전체 대문자로 바꿀 때

 

보통 컬럼명이 비슷하지만 대/소문자가 달라 merge가 되지 않는 경우가 존재했어요,

이럴 때 모두 대문자로 바꾸거나 소문자로 바꿔서 진행하는게 편한데

간단하게 List Comprehension을 이용해서 컬럼을 통일시켰습니다.

 

#2) 특정 혹은 전체 컬럼의 동일한 문자를 추가해야 하는 경우

# 특정 컬럼들을 추출해서 변경을 해야하는 경우
topic_dict = {x: x + '변경 문구' for x in df.columns if 특정 x를 추출할 조건}
df.rename(columns = topic_dict, inplace = True)

# 전체 컬럼에 대해 동일한 문자 추가를 해야하는 경우
df.columns = [x + '변경 문구' for x in df.columns]

 

특정 컬럼을 변경할 땐, 반복문을 사용해도 되지만 전 Dictionary Comprehension을 이용해서 수정했습니다.

시간도 많이 소요되지 않아 빠르고 간단합니다.

 

전체 컬럼에 특정 규칙을 적용한다면 List Comprehension으로 컬럼 추출 및 규칙 적용으로 쉽게 컬럼을 수정하고 있습니다.

 

여기까지 DataFrame 특정 컬럼명 수정, 전체 컬럼명 수정하는 방법을 정리했습니다 :)