python - 检查字符串是否在 Pandas 数据框中

我想查看数据框内特定列中是否存在特定字符串。

我遇到了错误

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)的 bool 值的指标向量

因此，您可以使用

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

或any()，如果您不在乎有多少条记录与您的查询匹配

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