python – 检查字符串是否在pandas数据帧中

我想看看我的数据帧中的特定列中是否存在特定字符串.

我收到了错误

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

import pandas as pd

BabyDataSet = [('Bob', 968), ('Jessica', 155), ('Mary', 77), ('John', 578), ('Mel', 973)]

a = pd.DataFrame(data=BabyDataSet, columns=['Names', 'Births'])

if a['Names'].str.contains('Mel'):
    print "Mel is there"

解决方法:

a [‘Names’].str.contains(‘Mel’)将返回一个大小为len(BabyDataSet)的布尔值的指示符向量

因此,你可以使用

mel_count=a['Names'].str.contains('Mel').sum()
if mel_count>0:
    print ("There are {m} Mels".format(m=mel_count))

或者任何(),如果您不关心有多少记录与您的查询匹配

if a['Names'].str.contains('Mel').any():
    print ("Mel is there")