[Python]Dataframe에서 like 검색-str.startswith() , str.contains()

Python Dataframe에서 str을 사용하면 문자열을 사용하기 쉽습니다. str을 사용해서 dataframe에서 SQL의 like search처럼 사용하는 방법을 알아보았습니다.

 

 

example 데이터프레임 을 살펴보겠습니다. CA_1부터 TX_2까지 있고 item_id도 종류별로 있습니다.

 

example 데이터

이때 store_id가 'CA' 인 것만 나타내보겠습니다.

 

example[example['store_id'].str.startswith('CA')]

 

store_id가 CA%인 것이 나왔습니다. 여기서 startswith는 앞 문자만 검색가능해 %CA% 같은 검색은 하지 못합니다.

Dataframe에서 CA% 검색

앞뒤 글자 상관없이 특정 단어가 필요한 것을 찾으려면 contains 함수를 사용하면 됩니다.

 

이번에는 item_id에서 BB글자가 들어간 것을 찾아보겠습니다.

 

example[example['item_id'].str.contains('BB')]

Dataframe에서 %BB% 검색

그럼 안녕