小编给大家分享一下怎么通过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来添加一列”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。