[pandas] 데이터 프레임의 인덱스에서 특정 값 제거하기

김판다t 2023. 6. 1. 16:07

데이터 프레임의 인덱스(index or columns)에서 특정 값을 제거하기

import pandas as pd
df = pd.DataFrame([1, 2, 3, 4, 5], index=list('ABCDE')).T

df :

df의 columns에서 'C'를 제거해 보자.

1. drop 함수로 해당 열을 삭제하고 columns를 반환한다.

df.drop('C', axis=1).columns

Index(['A', 'B', 'D', 'E'], dtype='object')

2. drop 함수는 인덱스에도 적용이 가능하다.

df.columns.drop('C')

3. 열 이름이 너무 길어서 위치로 삭제하고 싶다면 아래와 같은 코드로 가능하다.

df.columns.drop(df.columns[2])

4. 집합 연산으로 삭제할 수도 있다. (difference 함수 사용)

df.columns.difference(['C'])

difference 함수는 차집합 연산을 수행한다.

유튜브에서 판다스 강의 중입니다