时间戳

首页 > 解决方案 > 如果在数据框列中找到，则返回字符串的关键字

python - 如果在数据框列中找到，则返回字符串的关键字

问题描述

我有一个字符串。我需要查找该字符串中的关键字是否存在于我的数据框中。

如果存在，我需要返回该关键字。

细绳：

question="Joe is Available"
question=question.upper()
str_list=question.split()
str_list

输出[107]：

['JOE', 'IS', 'AVAILABLE']

数据框：

df=pd.DataFrame({"Person1":("Ash","Joe","Harry"),"Person2":("Abe","Lisa","Katty",),"Person3":("Sam","Max","Stone")})
df=df.apply(lambda x: x.astype(str).str.upper())


Person1 Person2 Person3
ASH     ABE     SAM
JOE     LISA    MAX
HARRY   KATTY   STONE

我的尝试：

return_field=""
for x in str_list:
    print(x)
    for i in df.iterrows():
        if(df.str.contains(x)):
            return_field=x

给我 AttributeError: 'DataFrame' 对象没有属性 'str'

预期产出

由于 Joe 存在于数据框中，它应该将我返回“Joe”

标签： pythonregexstringpandasdataframe

解决方案

如果您重复执行此操作，您可能希望通过set. 此外，您可以使用mapwithstr.upper将数据帧值转换为大写1：

str_all = set(map(str.upper, df.values.ravel()))

question = "Joe is Available"
str_search = set(question.upper().split())

res = str_search & str_all

# {'JOE'}

1您可以使用pd.DataFrame.apply+ lambda，但不建议这样做。字符串操作 viapd.Series.str目前是出了名的慢。lambda在顶部添加一个循环会使情况变得更糟。

推荐阅读