如果列包含任何指定的部分字符串，Pandas Dataframe保留行

Question

我有一个熊猫数据框。 下面是一个示例表。

Event   Text
A       something/AWAIT TO SHIP hello          
B       13579
C       AWAITING SHIP
D       24613
E       nan 

我只想在“文本”列中保留包含单词“ AWAIT TO SHIP”或在“文本”列中包含字符串13579或24613的行。 下面是我想要的表：

Event   Text
A       something/AWAIT TO SHIP hello          
B       13579
D       24613

下面是我尝试的代码：

df_STH001_2 = df_STH001[df_STH001['Text'].str.contains("AWAIT TO SHIP") == True | df_STH001['Text'].str.contains("13579") == True | df_STH001['Text'].str.contains("24613") == True]

以下是我得到的错误：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

Answer 1

您不应显式检查== True ，而应使用对contains的调用。

这是您的示例：

首先，我们定义样本数据框：

df1 = pd.DataFrame(data=[
('A', 'something/AWAIT TO SHIP hello'),
('B', 13579),
('C', 'AWAITING SHIP'),
('D', 24613),
('E', np.nan)], columns=['Event', 'Text'])

然后，我根据您的情况构建一个中间蒙版：

In [18]: mask = df1.Text.str.contains('AWAIT TO SHIP') |    \
                df1.Text.str.contains('13579') | \
                df1.Text.str.contains('24613')

现在，您可以使用此掩码为原始数据帧编制索引。

In [19]: df1.loc[mask]
Out[19]: 
  Event                           Text
0     A  something/AWAIT TO SHIP hello