利用Python實現(xiàn)數(shù)據(jù)匹配(vlookup)

平時我們經(jīng)常會制作周報、月報這種Excel操作相同的報表,我們可以使用python的pandas庫對其進行自動化。這篇文章就演示了利用python實現(xiàn)使用Excel中的vlookup函數(shù),并將其腳本封裝成exe可執(zhí)行程序。
寫這段腳本的原因是因為我們部門的一個做渠道的同事不太會用vlookup函數(shù),所以我琢磨著將他要用到vlookup的工作需求寫成腳本供他使用。
需求:要將B表中的“收入”字段的數(shù)據(jù)匹配到A表中去:
表A:


表A

表B


表B

結(jié)果:
結(jié)果

腳本內(nèi)容如下:

import pandas as pd

table_a_name = input("請輸入A表文件名:")
table_a_path = table_a_name + '.xlsx'
sheet_a_name = input("請輸入A表中的sheet名稱:")
table_a = pd.read_excel(table_a_path,sheet_name = sheet_a_name,converters={'訂單號':str}).dropna(axis=1,how='all')
table_b_name = input("請輸入B表文件名:")
table_b_path = table_b_name + ".xlsx"
sheet_b_name = input("請輸入B表中的sheet名稱:")
table_b = pd.read_excel(table_b_path,sheet_name=sheet_b_name,converters={'交易ID':str})
table_b_2 = table_b.groupby("交易ID").收入.sum().reset_index()
table_c = table_a.merge(right=table_b_2,how='left',left_on='訂單號',right_on='交易ID')
table_c.to_excel('c.xlsx',index=False)

考慮到同事不會使用python,故將這段腳本封裝成可執(zhí)行exe文件,關(guān)于怎樣將Python代碼封裝打包成可執(zhí)行的exe文件以及可能碰見的報錯解決詳見我之前的文章:將python腳本打包封裝成exe程序

過程演示.gif

但是上面的exe使用起來并不夠直觀、方便,故又利用tkinter制作了一個GUI界面:
GUI界面.gif

制作GUI界面的腳本如下:

#!/usr/bin/python
# -*- coding:utf-8 -*-

from tkinter import *
import pandas as pd
import os
import re
window = Tk()
window.title('數(shù)據(jù)匹配')
window.geometry('300x400')
var1 = StringVar()
var2 = StringVar()
var3 = StringVar()
var4 = StringVar()
var5 = StringVar()
var6 = StringVar()
var7 = StringVar()
L1 = Label( text="請輸入表A的名稱:")
L1.pack()
e1 = Entry(window,show=None,textvariable=var1)
e1.pack()
L2 = Label( text="請輸入表A中sheet名稱:")
L2.pack()
e2 = Entry(window,show=None,textvariable=var2)
e2.pack()
L3 = Label( text="請輸入表A中的ID列:")
L3.pack()
e3 = Entry(window,show=None,textvariable=var3)
e3.pack()
L4 = Label( text="請輸入表B的名稱:")
L4.pack()
e4 = Entry(window,show=None,textvariable=var4)
e4.pack()
L5 = Label( text="請輸入表B中sheet名稱:")
L5.pack()
e5 = Entry(window,show=None,textvariable=var5)
e5.pack()
L6 = Label( text="請輸入表B中的ID列:")
L6.pack()
e6 = Entry(window,show=None,textvariable=var6)
e6.pack()
L7 = Label( text="請輸入要匹配的表B中的列名:")
L7.pack()
e7 = Entry(window,show=None,textvariable=var7)
e7.pack()


def data_match():
    path = 'C:\\Users\\PS20190225\\Desktop'
    os.chdir(path)

    table_a_path = e1.get()
    a_type = re.search(r'([a-z]*).([a-z]*)',table_a_path).group(2)
    sheet_a_name = e2.get()
    A_id = e3.get()
    if a_type == 'xlsx':
        table_a = pd.read_excel(table_a_path,sheet_name = sheet_a_name,converters={A_id:str}).dropna(axis=1,how='all')
    else:
        table_a = pd.read_csv(table_a_path, sheet_name=sheet_a_name, converters={A_id: str}).dropna(axis=1, how='all')

    table_b_path = e4.get()
    b_type = re.search(r'([a-z]*).([a-z]*)',table_b_path).group(2)
    sheet_b_name = e5.get()
    B_id = e6.get()
    target_col = e7.get()
    if b_type == 'xlsx':
        table_b = pd.read_excel(table_b_path, sheet_name=sheet_b_name, converters={B_id: str}).dropna(axis=1, how='all')
    else:
        table_b = pd.read_csv(table_b_path, sheet_name=sheet_b_name, converters={B_id: str}).dropna(axis=1, how='all')
    table_b_2 = table_b.groupby(B_id)[target_col].sum().reset_index()

    table_c = table_a.merge(right=table_b_2,how='left',left_on=A_id,right_on=B_id)
    table_c.to_excel('c.xlsx',index=False)

b = Button(window,text='進行匹配',width=15,height=2,command=data_match)
b.pack()
window.mainloop()

學習tkinter可觀看莫凡大佬的視頻:做簡單窗口視窗

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • 平時我們經(jīng)常會使用Excel中的vlookup函數(shù),如果我們有很長一段時間重復相同操作我們可以使用Python將其...
    小T數(shù)據(jù)站閱讀 2,982評論 0 2
  • 一、Python簡介和環(huán)境搭建以及pip的安裝 4課時實驗課主要內(nèi)容 【Python簡介】: Python 是一個...
    _小老虎_閱讀 6,356評論 0 10
  • 1) 那年,我三個月。 媽媽走了。 2) 那年,我6歲。 媽媽已經(jīng)好久沒回來了,媽媽過年就回來了,我不喜歡媽媽,我...
    壹狗閱讀 449評論 0 4
  • 6 焦鵬的爸爸吃了一驚,沒想到會有一個女人沖過來,“你是誰?”他錯愕地盯著金香蘭,打人的手掌在空中劃著弧線不知該怎...
    雅賢閱讀 581評論 0 3
  • 組別 301期 利他一組 【日精進打卡第090天】 【知~學習】 誦讀《六項精進大綱》3遍,累計208遍;誦讀《...
    J0hn先生閱讀 170評論 0 0

友情鏈接更多精彩內(nèi)容