본문 바로가기
Python

[pandas] Column (or rows) 제거하기

by ds31x 2024. 1. 9.

DataFrame 에서 column을 제거하는데에 사용되는 idiomatic approach는 drop 메서드를 사용하는 것임.

사실 dropaxis라는 parameter를 가지고 있고,

  • 0이 주어지면 row를 지우고,
  • 1이 주어지면 column을 지움.

첫번째 argument로

  • 지우고자 하는 column의 이름을 주거나
  • 해당 DataFrame의 instance 의 columns 어트리뷰트에서 index로 지정하거나 : a.columns[2]
  • list로 지우고자 하는 columns의 이름들을 주면 됨.

다음의 예제 코드를 참고하라.

import pandas as pd

a = pd.DataFrame({'name': ['kim hk', 'lee mb', 'tae yk'], 
                  'age': [33,22,12], 
                  'weight': [72,80,50]})

b = a.drop( ['age'], axis=1 )
b.head(3)

참고 : idiomatic method.

여기서 idiomatic method 란 관용적인 사용이라고도 불리는 용어로,

특정 프로그래밍 언어나 라이브러리 등에서 자연스럽게 일반적으로 사용되는 방법을 의미함.

예를 들어, 문을 여는 idiomatic method는 문손잡이를 손으로 돌려 여는 것이라 할 수 있음

(다르게 여는 방법은 문손잡이를 입이나 발을 이용할 수도 있지만 자연스럽진 않음)