[python] 데이터 프레임 열 순서 변경하는 방법

 

 

데이터 프레임(DataFrame) 열 순서 변경하는 방법

 

데이터 프레임에서 열 끼리의 연산을 해서 새로운 열을 만들어 냈는데 그 경우 열이 맨 끝에 추가되서 보기가 어렵다거나 그냥 열 순서가 엉망인 경우... 열 순서를 바꾸고 싶은 경우가 있다.

 

열 순서를 변경하는 방법은 간단하다.

 

'이름', '성별', '나이' 라는 세 개의 열을 가진 df 라는 데이터프레임이 있을 때

 

df = df[['성별', '나이', '이름']]

 

이렇게 원하는 순서대로 열 이름을 리스트로 넣어주면 된다.

모든 열의 이름을 넣어야 한다.

 

만약 ['성별', '이름'] 이렇게 넣게 되면 '나이' 열은 없어지게 된다.

열을 삭제하고 싶었다면 그렇게 해도 된다.

 

 

예시)

 

import numpy as np
import pandas as pd
import seaborn as sns # 예시로 seaborn 기본 데이터셋 사용할 거라서 필요

df = sns.load_dataset('exercise')

df.columns # 데이터 프레임의 열 확인
# Index(['Unnamed: 0', 'id', 'diet', 'pulse', 'time', 'kind'], dtype='object')
# 6개의 열이 있음

# 'kind' 라는 열을 맨 앞으로 가져오고 싶다면?
df = df[['kind', 'Unnamed: 0', 'id', 'diet', 'pulse', 'time']]
df

 

기본 - 바꾸기 전

 

kind 열을 맨 앞으로 가져온 경우

 

열 순서가 잘 바뀐 것을 확인할 수 있다.

 

---

 

+) 열을 삭제하는 방법

 

# 'Unnamed: 0' 열이 필요없는 것 같아서 빼버리고 싶다면?

df = df[['kind', 'id', 'diet', 'pulse', 'time']]
df

 

 

 

데이터 프레임에 리스트 형식으로 원하는 column 만 적으면 원하는 열만 저장되어 나오게 된다.