我正在尝试使用
str.contains
函数在python中搜索列中的“关键字”。
我已经成功地在一列中查找了一个关键字。
但是,我需要的是:
-在一列中搜索多个关键字
我正在使用以下示例数据集:
Employee Employee ID Category Comments
Jack 11234 Holiday Holiday trip
Ben 1921 Business Business trip
John 233535 Other Other trip
Pete 892230 Other Other trip
使用以下代码:
df[df['Category'].str.contains("holiday", case=False)]
我已成功获得以下结果:
Employee Employee ID Category Comments
Jack 11234 Holiday Holiday trip
但是,我想搜索关键字假日或商务。使用“或”运算符,似乎无法使其工作。有什么关于扩展代码的建议吗?这样我就可以同时搜索“假日”和“商务”两个词了。
最终结果将需要如下所示(即返回列类别中具有关键字holiday或business的行):
Employee Employee ID Category Comments
Jack 11234 Holiday Holiday trip
Ben 1921 Business Business trip
下一部分是我要在多个列中搜索关键字。我尝试了以下代码:
df[df['Category'].str.contains("holiday", case=False)] |
df[df['Comments'].str.contains("holiday", case=False)]
但不幸的是,这并没有产生预期的结果。
非常感谢您的洞察力!