[Python] 데이터프레임 컬럼 삭제 방법

안녕하세요! 데이터프레임의 컬럼 삭제 방법에 대해 알아보도록 하겠습니다.

데이터프레임의 컬럼 삭제

데이터프레임에서 컬럼을 삭제하는 방법에는 여러 가지가 있습니다. 여기서는 파이썬의 pandas 모듈을 사용하는 방법에 대해 설명하겠습니다.

1. del 문 사용하기

가장 간단한 방법은 del 문을 사용하는 것입니다. 다음과 같이 del 문과 데이터프레임의 컬럼 이름을 사용하여 컬럼을 삭제할 수 있습니다.

 

import pandas as pd

 

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

del df['B']

print(df)

 

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

   A  C
0  1  7
1  2  8
2  3  9

2. drop() 함수 사용하기

pandas 모듈에서는 drop() 함수를 사용하여 컬럼을 삭제할 수도 있습니다. drop() 함수는 데이터프레임에서 지정한 축을 따라 행이나 열을 삭제합니다. 축을 지정하지 않으면 행을 삭제합니다.

 

import pandas as pd

 

f = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

df = df.drop('B', axis=1)

print(df)

 

위 코드를 실행하면 결과는 del 문을 사용한 예제와 동일합니다.

   A  C
0  1  7
1  2  8
2  3  9

3. pop() 함수 사용하기

pop() 함수는 데이터프레임에서 컬럼을 삭제하면서 해당 컬럼의 값을 반환합니다. 다음과 같이 pop() 함수를 사용하여 컬럼을 삭제할 수 있습니다.

 

import pandas as pd

 

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

df.pop('B')

print(df)

 

위 코드를 실행하면 결과는 del 문과 drop() 함수를 사용한 예제와 동일합니다.

   A  C
0  1  7
1  2  8
2  3  9

마무리

이상으로 데이터프레임에서 컬럼을 삭제하는 방법에 대해 알아보았습니다. 각각의 방법은 상황에 따라 다르게 사용될 수 있으니 필요에 따라 적절한 방법을 선택하여 사용하시기 바랍니다.