본문 바로가기

BLOG/파이썬 기초

[파이썬] 데이터프레임(dataframe) 지정된 행 삭제하기 - drop()

데이터프레임을 이용하여 데이터를 다룰 때, 특정 행을 지워야 할 수 있는데요,

이때 사용하는 함수가 바로 drop() 입니다.

drop() 함수는 .drop([행번호]) 와 같이 사용할 수 있습니다.

 

예를 들어, 아래와 같은 데이터프레임이 있다고 가정해보겠습니다.

 

import pandas as pd
info = pd.DataFrame({'item' : ['coffee', 'tea', 'juice' ], 
                     'price' : [3000, 4000, 5000], 
                     'rating' : [4.0, 3.5, 3.7]})
info

 

	item	price	rating
0	coffee	3000	4.0
1	tea	4000	3.5
2	juice	5000	3.7

 

위 데이터프레임에서 만약 인덱스가 1인 'tea'-'4000'-'3.5' 행을 지우고 싶다면 아래와 코드를 작성하면 됩니다.

 

info.drop([1])

 

	item	price	rating
0	coffee	3000	4.0
2	juice	5000	3.7

 

출력된 결과를 보면 첫 번째 행이 잘 삭제되었음을 확인할 수 있습니다.