怎么通过pandas column内容匹配字典的key来添加一列

小编给大家分享一下怎么通过pandas column内容匹配字典的key来添加一列，相信大部分人都还不怎么了解，因此分享这篇文章给大家参考一下，希望大家阅读完这篇文章后大有收获，下面让我们一起去了解一下吧！

这是知乎上的问题，原问题是这样的：

我想根据H列的交易对方的关键字，生成Q列类型。比如H列第一个包含“美的“，那Q列就是"电器"。

如果一个一个添加我会，我想生成一个字典{"官方":"电器"，“网络技术”:"投资"……}，通过匹配dataframe中H列内容包含字典key，来添加一列，匹配类型。请问如何操作？

我来回答一下：

''总体思路是利用遍历字典的key值，然后使用in判断，如果true则返回value'''
# 构造的字典
>>> my_dict ={'bo': 'A', 'jik': 'B', 'luc': 'C'}
# 构造的dataframe
>>> df = pd.read_excel(r'D:/myExcel/1.xlsx')
>>> df
    name  score
0    bob     45
1  jiken     67
2    bob     23
3    bob     34
4   lucy     42
5   lucy     75
# 定义函数，进行逻辑判断
>>> def get_res(strings):
    for key, value in my_dict.items():
        if key in strings:
            return value
    return ""
# 对name的每一行运用apply函数，获取返回结果，并返回grade列
>>> df['grade'] = df['name'].apply(lambda x: get_res(x))
>>> df
    name  score grade
0    bob     45     A
1  jiken     67     B
2    bob     23     A
3    bob     34     A
4   lucy     42     C
5   lucy     75     C

以上是“怎么通过pandas column内容匹配字典的key来添加一列”这篇文章的所有内容，感谢各位的阅读！相信大家都有了一定的了解，希望分享的内容对大家有所帮助，如果还想学习更多知识，欢迎关注亿速云行业资讯频道！