python实现vlookup功能_Python实现excel vlookup功能

目标：

按Chapter关键字合并2个表，并且保存为genSum.csv

实现思路：

csv表中Name中先提取关键字copy到Chapter列。然后merger表2，最后保存为genSum.csv。

截图.png

实现代码：

__author__ = 'cllea'

import pandas as pd

import numpy as np

df = pd.read_csv("task.csv")

#提取Name列

s =df["Name"]

#转为list

listName=s.tolist()#list

#在list中修改字符串

for i, v in enumerate(listName):

listName[i] = v.strip()[v.index(']')+2:v.index(']')+11]

#print(listName)

#list转为dataframe

data = pd.DataFrame(listName,columns=['Chapter'])

#print(data)

#按列拼接dataframe

dfA=pd.concat([df,data],axis=1)

#print(dfA)

#合并dataframe

dfB = pd.read_excel("myplan.xlsx")

#print(dfB)

#对关键字Chapter列向左连接（左边dfA为全部）

dfC = pd.merge(dfA, dfB,how='left',on=['Chapter'])

#print(dfC)

#保存到csv中

dfC.to_csv('genSum.csv',chunksize=10,encoding="utf_8_sig")