python不到50行代碼完成了多張excel合併

一 前言

公司同事最近在做excel相關的工作;今天來求助知識追尋者合併多個excel爲一個一個工作本,原本是java操作poi太蛋疼了,笨重不堪,內存消耗嚴重,知識追尋者使用python不到40行代碼完成了60多張excel工作本合併爲一張;python真香

牛皮吹完了,如果看過知識追尋者系列文章的讀者肯定知道之前知識追尋者發過一篇

python專題使用openpyxl操作excel(公衆號讀者直接在彙總欄查閱);本篇使用的不是openpyx庫,使用的使是

xlrd,xlwt庫,雖然這兩庫功能沒法根openpyx相比,但可以操作xls結尾的舊版excel而openpyx不支持;

知識追尋者(Inheriting the spirit of open source, Spreading technology knowledge;)

二 代碼

大體思路如下

  • 遍歷獲取根目錄下的所有excel文件
  • 根據excel名稱進行匹配獲取某類excel
  • 創建工作本用於寫入拷貝的數據
  • 每個excel都有一張Sheet1,循環遍歷單元格寫入創建的工作本

詳細過程寫在代碼裏面;讀者可以參考一下思路和使用方式,後續知識追尋者有空會出這2庫基礎系列文章的使用方式;

# -*- coding: utf-8 -*-
import xlrd
import xlwt
import os
import re

""" 寫入工作本 """
def write_excel(path, write_sheet):
    # 加載工作本
    book = xlrd.open_workbook(path)
    # 獲取表單
    read_sheet = book.sheet_by_name('Sheet1')
    # 遍歷
    for row in  range(read_sheet.nrows):
        for col in  range(read_sheet.ncols):
            write_sheet.write(row, col, read_sheet.cell_value(row,col))
            
# 獲取根目錄下所有文件名
def walk(path):
  for root,dirs,names in os.walk(path):
    list = []
    for filename in names:
        path = os.path.join(root, filename)
        list.append(path)
    return list

if __name__ == "__main__":
    # 創建工作本
    write_book = xlwt.Workbook()
    # 根目錄
    root = r'C:\mydata\generator\excel'
    path_list = walk(root)
    for path in path_list:
        val = path.find("本專科")
        if val!=-1:
        	# 正則匹配
            ser = re.search('.*20200403(.*?).xls',path)
            name = ser.group(1)
            # 創建sheet
            write_sheet = write_book.add_sheet(name)
            # 寫入
            write_excel(path, write_sheet)
    # 保存        
    write_book.save(r'本專科.xls')
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章