python - 找出数据框列是否包含单词 python

Question

我有dataframe这样的

 name    cat1    cat2    cat3
'aa bb'    A      A-1   A-1-1
'cc dd'    B      B-1   B-1-1
'ee aa'    C      C-1   C-1-1
'gg bb'    D      D-1   D-1-1

我想制作dataFrame包括dataframe这样aa的内容

 name    cat1    cat2    cat3
'aa bb'    A      A-1   A-1-1
'ee aa'    C      C-1   C-1-1

Answer 1

使用strpandas 系列的方法进行字符串操作，包括检查字符串是否包含某个子字符串：

df.loc[df.name.str.contains("aa")]