[pandas] 모든 셀에서 찾고자 하는 string이 있는 row 찾기

다음과 같은 dataframe이 있다고 하자.

import pandas as pd
import numpy as np


df = pd.DataFrame(
    {
        'Zodiac' : ['Libra','Capricorn','Scorpio','January','Scorpio', 'Capricorn','Capricorn', 'Taurus'],
        'Month' : ['January', 'February', 'February', 'April', 'January', 'March', 'April', 'April'],
        'values' : [23, 45,89, 91, "january", "february", "april", "january"]
    }
)

df

>>>>

 

이때 어떤 컬럼도 상관없이 단어 Jan을 포함하고 있는 row들을 찾고한다.

다음과 같이 str.contains()을 활용하면 된다.

substring = 'jan'
df[df.apply(lambda row: row.astype(str).str.contains(substring, case=False).any(), axis=1)]

>>>>

 

 

만약 대소문자를 구분해서 찾고 싶다면 str.contains(substring, case=True)로 찾으면 된다.

substring = 'jan'
df[df.apply(lambda row: row.astype(str).str.contains(substring, case=True).any(), axis=1)]

 

 

 

[참고자료]

• https://kanoki.org/2022/02/04/pandas-search-a-string-in-dataframe-across-all-columns/