[Pandas] 데이터프레임 컬럼 순서 변경하기

[Pandas] 데이터프레임 컬럼 순서 변경하기

데이터 프레임을 다루다보면, 데이터 프레임의 컬럼 순서를 변경해야하는 경우가 종종 있다.

이번 포스팅에선 데이터프레임의 컬럼 순서를 변경하는 몇 가지 방법에 대해 알아보고자 한다.

 

샘플 데이터 프레임 생성

먼저 다음과 같이 데이터 프레임을 생성해준다.

import pandas as pd 

data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}

df = pd.DataFrame(data)

 

reindex

`reindex` 메소드를 사용하면 원하는 순서로 컬럼을 재배치 할 수 있다.

new_order = ['C', 'A', 'B']

df_reordered = df.reindex(columns=new_order)

 

직접 인덱싱

데이터 프레임의 컬럼을 리스트로 직접 인덱싱하여 순서를 변경할 수 있다.

df_reordered = df[['C', 'A', 'B']]

 

INSERT 메서드 사용

특정 컬럼을 원하는 위치에 삽입할 수도 있다. 아래 예시에서는 'C' 컬럼을 첫 번째 위치에 삽입 한다.

df_reordered = pd.DataFrame()
df_reordered.insert(0, 'C', df['C'])
df_reordered.insert(1, 'A', df['A'])
df_reordered.insert(2, 'B', df['B'])

 

POP 메서드와 INSERT 조합

`pop`메서드를 사용하여 특정 컬럼을 제거한 후, `insert`로 원하는 위치에 다시 추가하는 방식이다.

c_column = df.pop('C')
df.insert(0, 'C', c_column)

 

iloc 사용

숫자 인덱스를 이용하여 컬럼의 순서를 변경할 수도 있다.

해당 방법은 컬럼의 위치를 알고 있을 때 유용하다.

df_reordered = df.iloc[:, [2, 0, 1]]  # C, A, B 순서