温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

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

发布时间:2021-11-30 10:34:37 来源:亿速云 阅读:377 作者:小新 栏目:大数据

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

这是知乎上的问题,原问题是这样的:

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

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

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

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI