[DataFrame] 특정 문자와 일치하는 행 추출 및 제거하기

해당 데이터는 공공데이터 포털 사이트에서 가져온 데이터입니다.

자주 사용하는데 자꾸 까먹어서 쓸 때마다 인터넷에 구글링하는 시간들이 아까워서 정리해놓으려 한다..

 

추출

'신호등 종류'라는 컬럼에서 '보행등'과 일치하는 행만 추출하고자한다면!

한 줄로 해결 완료!

 

 

삭제

데이터 프레임에서 특정 문자열이 있는 행을 삭제하고자 할 때는 위에서 추출한 행들의 index를 이용한다.

 

아래의 코드는,

데이터프레임의 '신호등 종류'라는 컬럼에서 '버스'라는 문자열 값이 있는 행들만 추출하여

그 행들의 index들을 이용해서! drop 함수로 제거하는 내용이다. 

# idx는 '버스'와 일치하는 문자열인 행들의 인덱스 번호가 저장된다.
idx = traffic_light[traffic_light['신호등 종류'] == '버스'].index

traffic_light = traffic_light.drop(idx)

 

 

전체 코드를 보자면,

idx, idx2, idx3, idx4에 넣고, drop을 한 것이다.

 

+ 지금 보니, 저 drop이나 index 저장하는 명령어도 for문으로 돌리면 더 간결하겠다..