python - 使用宽松字典映射列中的值

Question

我想知道是否有人可以帮助我将一列字符串与 python 中的一种轻松字典匹配。所以我有以下 python 数据框:

String                      Colour
8392apple8309
8dbsfhorange9anld
38banananflks9

和这本字典:

_dict
{'apple':'Red',
'orange':'Orange'
'banana':'Yellow'}

我写了这个函数:

def fruitsearch(string):
    return [value for key, value in _dict.items() if string in key.lower()]

它能够获取我的键的子字符串，例如 fruitsearch('app') 并返回正确的颜色，红色。然而，我想要做的是让函数在我的数据框列“字符串”中找到键，并将正确的颜色返回到数据框的第二列颜色，所以它看起来像这样:

String                        Colour
8392apple8309                 Red
8dbsfhorange9anld             Orange
38banananflks9                Yellow

谢谢!

Answer 1

这是一种方法。

演示:

import pandas as pd


def fruitsearch(string):
    _dict = {'apple':'Red', 'orange':'Orange', 'banana':'Yellow'}
    for key, value in _dict.items():
        if key.lower() in string:
            return value 
    return None


df = pd.DataFrame({"String": ["8392apple8309", "8dbsfhorange9anld", "38banananflks9"]})
df["Colour"] = df["String"].apply(fruitsearch)

print(df)

输出:

              String  Colour
0      8392apple8309     Red
1  8dbsfhorange9anld  Orange
2     38banananflks9  Yellow