본문 바로가기
5-3. Pandas 제공 자료구조/DataFrame

[파이썬 판다스] 데이터프레임에서 특정 열 제거하기

by 만다린망고 2021. 12. 30.
반응형

판다스를 불러옵니다. 

import pandas as pd


먼저 딕셔너리를 정의합니다. 

dic={'이름':['이호정','이정진','이기호','정희진','이나리'],
        '성별':['남','남','남','여','여'],
        '나이':[25,45,72,25,37]
       }


딕셔너리를 이용하여 데이터프레임을 정의합니다.

data1=pd.DataFrame(dic)

>>> data1
    이름 성별  나이
0  이호정  남  25
1  이정진  남  45
2  이기호  남  72
3  정희진  여  25
4  이나리  여  37


성별 열을 제거해봅시다. drop 메소드를 사용합니다. 행을 제거할 때와 달리 옵션을 입력해야 합니다. axis=1 또는 axis='columns' 옵션을 입력합니다. 

>>> data1.drop('성별',axis=1)
    이름  나이
0  이호정  25
1  이정진  45
2  이기호  72
3  정희진  25
4  이나리  37


리스트를 이용해서 여러 행을 제거하는 것도 가능합니다. 

주의할 특징은 원래 데이터가 제거되지는 않는다는 것입니다. 제거된 데이터프레임을 반환하는 것이지 원래 데이터에서 특정 행을 제거하지는 않습니다. 

>>> data1
    이름 성별  나이
0  이호정  남  25
1  이정진  남  45
2  이기호  남  72
3  정희진  여  25
4  이나리  여  37


원래 데이터에서까지 제거하려면 inplace=True 라는 옵션을 입력하면 됩니다. 

반응형

댓글