Pandas如何实现vlookup功能？

前端皮皮

发布于 2022-08-17 18:41:47

1.1K0

发布于 2022-08-17 18:41:47

文章被收录于专栏：前端进阶学习交流

大家好，我是皮皮。

一、前言

前几天在Python黄金交流群【Edward】问了一道Pandas处理的问题，如下图所示。

二、实现过程

方法一

这里【格格物 এ คิดถึง】给出了一个思路和代码。具体思路如下图所示：

代码如下图所示：

下图是运行结果：

可以解决粉丝的问题。

方法二

后来【月神】也给了一个方法，代码如下所示：

## merge 用法
regon_code = '3202'

dfb['大类编码'] = regon_code + dfb['大类编码']
dfb['小类编码'] = regon_code + dfb['小类编码']
# 这里已经merge好了
dfn = pd.merge(dfn, dfb, how='inner').rename(columns={'大类编码': '项目大类编码','小类编码': '项目小类编码'})

dfn.insert(0, '项目大类编码', dfn.pop('项目大类编码'))
dfn.insert(2, '项目小类编码', dfn.pop('项目小类编码'))
print(dfn)

方法三

后来【月神】还给了一个map方法，代码如下所示：

# map 用法
regon_code = '3202'

dfb['大类编码'] = regon_code + dfb['大类编码']
dfb['小类编码'] = regon_code + dfb['小类编码']
# map需要对映射表做去重处理
dfn.insert(0, '项目大类编码', dfn['项目大类'].map(dfb.set_index('项目大类')['大类编码'].drop_duplicates()))
dfn.insert(2, '项目小类编码', dfn['项目小类'].map(dfb.set_index('项目小类')['小类编码'].drop_duplicates()))

print(dfn)

【月神】对Pandas的掌握可谓出神入化，令人望尘莫及。