如何在pandas数据框架列中搜索单词或字符串?

Question

我有以下示例数据框架:

我想只选择列"中的行。有'SENSOR'这个词，大写或小写。

import pandas as pd
data = {'first_column': ['SEN_strainG', 'second_valueSEN', 'HHHH', 'sen_precipitacion', 'sen'],
'second_column': [1,2,3,4,5]}
df = pd.DataFrame(data)

我用下面的代码解决了这个问题，这个代码工作得很好:

df_result = df[(df['first_column'] == 'SEN_strainG') |
(df['first_column'] == 'second_valueSEN') |
(df['first_column'] == 'sen_precipitacion') |                  
(df['first_column'] == 'sen')]

然而，我认为有可能概括实现来寻找'sen'或'sen'。有没有一种方法可以做到这一点，并得到同样的结果?

Answer 1

df[df.first_column.str.contains('sen', na=False, flags=re.IGNORECASE)]

输出:

first_column  second_column
0        SEN_strainG              1
1    second_valueSEN              2
3  sen_precipitacion              4
4                sen              5