Pandas DataFrame에서 열 삭제하기

Pandas DataFrame에서 열을 삭제하는 방법은 두 가지가 있습니다.

• drop() 메서드
• pop() 메서드

이 글에서는 drop() 메서드를 활용하여 열을 삭제하는 방법에 대해서 설명드리겠습니다. 

 

drop() 메서드

df.drop(labels=None, axis=0, index=None, columns=None, level=None, inplace=False, errors='raise')

• labels: 삭제할 열 이름입니다. axis를 지정해주어야 합니다.
• axis: {0: index / 1: columns} labels 인수를 사용할 경우 지정할 축입니다.
• index: 인덱스명을 입력해서 바로 삭제를 할 수 있습니다.
• columns: 컬럼명을 입력해서 바로 삭제를 할 수 있습니다.
• level: 멀티인덱스의 경우 레벨을 지정해서 진행할 수 있습니다.
• inplace: 원본을 변경할지 여부입니다. True일 경우 원본이 변경됩니다.
• errors: 삭제할 열을 찾지 못할 경우 오류를 띄울지 여부입니다. ignore할 경우 존재하는 열만 삭제됩니다.

예시

다음과 같이 간단한 3x3 객체를 만들어 보겠습니다.

df = pd.DataFrame({
    'col1': [1, 2, 3],
    'col2': [4, 5, 6],
    'col3': [7, 8, 9]
})

 

axis=0과 labels를 사용하여 col2 열을 삭제해 보겠습니다.

df.drop(labels='col2', axis=0)

     col1  col3
0    1     3
2    4     9

 

index를 사용하여 row2 행을 삭제해 보겠습니다.

df.drop(index='row2')

     col1  col2
0    1     2
1    4     5

 

columns를 사용하여 col1 열을 삭제해 보겠습니다.

df.drop(columns='col1')

   col2  col3
0     2     3
1     5     6
2     8     9

 

결론

drop() 메서드를 사용하여 Pandas DataFrame에서 열을 삭제할 수 있습니다. 삭제할 열이나 행을 지정할 수 있으며, axis 인수를 사용하여 열이나 행을 지정할 축을 지정할 수 있습니다. 또한, inplace 인수를 사용하여 원본 DataFrame을 변경할 수 있습니다.