[PYTHON/PANDAS] DataFrame 클래스 : drop 메소드에서 axis/inplace 인자를 사용해 컬럼 삭제하기

■ DataFrame 클래스의 drop 메소드에서 axis/inplace 인자를 사용해 컬럼을 삭제하는 방법을 보여준다.

▶ main.py


import pandas as pd

dataFrame = pd.DataFrame(
     {
          "name"  : ["김철수", "이영희", "박민수", "정지은"],
          "age"   : [25, 30, 35, 28],
          "city"  : ["서울", "부산", "서울", "대전"],
          "score" : [85, 92, 78, 95]
     }
)

print(dataFrame)

"""
  name  age city  score
0  김철수   25   서울     85
1  이영희   30   부산     92
2  박민수   35   서울     78
3  정지은   28   대전     95
"""

print()

dataFrame.drop(["city", "score"], axis = 1, inplace = True) # axis : 0(행), 1(컬럼)

print(dataFrame)

"""
  name  age
0  김철수   25
1  이영희   30
2  박민수   35
3  정지은   28
"""

import pandas as pd

dataFrame = pd.DataFrame(

{

"name" : ["김철수", "이영희", "박민수", "정지은"],

"age" : [25, 30, 35, 28],

"city" : ["서울", "부산", "서울", "대전"],

"score" : [85, 92, 78, 95]

}

)

print(dataFrame)

"""

name age city score

0 김철수 25 서울 85

1 이영희 30 부산 92

2 박민수 35 서울 78

3 정지은 28 대전 95

"""

print()

dataFrame.drop(["city", "score"], axis = 1, inplace = True) # axis : 0(행), 1(컬럼)

print(dataFrame)

"""

name age

0 김철수 25

1 이영희 30

2 박민수 35

3 정지은 28

"""

▶ requirements.txt


numpy==2.1.2
pandas==2.2.3
python-dateutil==2.9.0.post0
pytz==2024.2
six==1.16.0
tzdata==2024.2

numpy==2.1.2

pandas==2.2.3

python-dateutil==2.9.0.post0

pytz==2024.2

six==1.16.0

tzdata==2024.2

※ pip install pandas 명령을 실행했다.