Python Excel数据匹配 利用Python实现Excel的文件间的数据匹配功能
chaodaibing 人气:2我们知道Excel有一个match函数，可以做数据匹配。
比如要根据人名获取成绩
而参考表sheet1的内容如下：
要根据sheet1匹配每人的成绩，用Excel是这么写
index(Sheet1!B:B,MATCH(A2,Sheet1!A:A,0))
意思就是获取sheet1的B列的内容，根据我的A列匹配sheet1的A列的内容
但是如何用python实现这一点呢，我写了一个函数，非常好用，分享给大家。
这个函数考虑到了匹配多个字段，多个sheet。
import pandas as pd
def match(file,sheetnames,reffile,refsheet,targetsegs,matchseg) #文件名 sheet列表 参考文件名 参考sheet 目标字段列表 参考字段
alldata=pd.read_excel(file,None)
refdata=pd.read_excel(reffile,refsheet)
#获取映射字典
maps={}
for i in refdata.index:
MatchSeg=refdata.loc[i,matchseg]
maps[MatchSeg]={}
for seg in targetsegs:
maps[MatchSeg][seg]=refdata.loc[i,seg]
#匹配数据
for sheet in sheetnames:
if(isinstance(sheet,int)):
sheet=list(alldata.keys())[sheet]
data=alldata[sheet].fillna('-')
for i in data.index:
MatchSeg=data.loc[i,matchseg]
for seg in targetsegs:
try:
data.loc[i,seg]=map[MatchSeg][seg]
except Exception as e:
pass
alldata[sheet]=data
#导出
with pd.ExcelWriter(file) as writer:
for sheet in alldata.keys():
alldata[sheet].to_excel(writer,sheet,index=False)
match('要匹配的表.xlsx',[0,1],'参考表.xlsx','参考页',['要匹配的字段1,字段2'],'参考字段')
总结
加载全部内容