千家信息网

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

发表于:2024-10-09 作者:千家信息网编辑
千家信息网最后更新 2024年10月09日,小编给大家分享一下怎么通过pandas column内容匹配字典的key来添加一列,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一
千家信息网最后更新 2024年10月09日怎么通过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来添加一列"这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注行业资讯频道!

0