파이썬 Pandas DataFrame에서 열(컬럼) 삭제하는 방법

Pandas DataFrame에서 특정 열(column)을 삭제하는 가장 간단한 방법은 del 키워드를 사용하는 것입니다. 이 외에도 pop() 메서드나 drop() 메서드를 활용하면 상황에 맞게 유연하게 열을 제거할 수 있습니다.

del 키워드로 열 삭제하기

del 키워드 뒤에 대괄호([])를 붙이고 그 안에 삭제하고 싶은 열의 이름을 지정하면 해당 열이 DataFrame에서 즉시 제거됩니다.

del dataFrame['ColumnName']

1. 라이브러리 임포트

먼저 pandas 라이브러리를 별칭(alias)과 함께 불러옵니다.

import pandas as pd

2. DataFrame 생성

예제로 사용할 DataFrame을 만들어 보겠습니다.

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

3. 'Car' 열 삭제

이제 del 키워드를 사용해 'Car' 열을 삭제합니다.

del dataFrame['Car']

전체 예제 코드

지금까지의 과정을 하나의 코드로 정리하면 다음과 같습니다.

import pandas as pd

# DataFrame 생성
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print("DataFrame ...\n", dataFrame)

# 열 삭제
del dataFrame['Car']

print("\n열 삭제 후 DataFrame = \n", dataFrame)

실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

DataFrame ...
      Car  Units
0     BMW    100
1   Lexus    150
2    Audi    110
3 Mustang     80
4 Bentley    110
5  Jaguar     90

열 삭제 후 DataFrame =
   Units
0   100
1   150
2   110
3    80
4   110
5    90

참고: pop()과 drop() 메서드

pop() 메서드는 열을 삭제하면서 동시에 삭제된 열의 데이터를 반환합니다. 단, 원본 DataFrame이 직접 수정된다는 점에 유의해야 합니다.

units = dataFrame.pop('Units')  # 열을 삭제하고 해당 열의 데이터를 변수에 저장

drop() 메서드는 여러 열을 한 번에 삭제할 수 있으며, inplace=True 옵션을 지정하지 않으면 원본을 유지한 채 새로운 DataFrame을 반환하므로 실무에서 더욱 안전하게 활용됩니다.

result = dataFrame.drop('Car', axis=1)        # 새로운 DataFrame 반환
dataFrame.drop('Car', axis=1, inplace=True)   # 원본에서 직접 삭제