熊猫-搜索关键词

Question

我试图在dataframe中搜索关键字，如果使用以下代码找到关键字，则打印关键字：

if df[df['description'].str.contains(keyword,case=False)]:
    print(keyword)
else:
    print("NOT FOUND") 

我收到以下错误消息：

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

对如何解决这个问题有什么建议吗？

Answer 1

试一试：

if df['description'].str.contains(keyword,case=False).any():
    print(keyword)
else:
    print("NOT FOUND") 

df['description'].str.contains(keyword,case=False)返回一个列，其长度与df['description']相同，包含True或False值，其中每个值对应于df['description']中相同索引的值。如果该值包含keyword，则返回的系列中的值为True，否则为False。

如果在Series对象中至少有一个值是.any()，则调用Series对象上的True将返回True。如果它们都是False，.any()将返回False。

Answer 2

你的表情

df[df['description'].str.contains(keyword,case=False)]

不返回bool，它返回原始dataframe的子集，其中包含与谓词匹配的行(contains关键字)。因此，正如错误消息所暗示的，您需要测试此数据是否为空。