快捷導航

python實現(xiàn)自動化之文件合并

更新時間：2021年08月12日 10:28:31 作者：Alan.hsiang

這篇文章主要為大家詳細介紹了python實現(xiàn)自動化文件合并，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

假如公司需要統(tǒng)計每個員工的個人信息，制定好模板后，由員工填寫，然后發(fā)送到綜合部進行匯總，在這種情況下，如果公司有上百位員工的信息需要統(tǒng)計，且采用純手工進行復制粘貼的方式進行匯總，則將是一項耗時費力易錯的工作。本文主要以一個簡單的小例子，簡述如何通過Pyhton進行自動化文件合并。

涉及知識點

xlrd模塊：文件的讀取
xlwt模塊：文件的生成

設(shè)計思路

1、獲取指定目錄下所有excel格式的文件。
2、循環(huán)讀取Excel文件內(nèi)容【讀取固定的行與列】，將內(nèi)容暫存內(nèi)存中。
3、將內(nèi)存中數(shù)據(jù)，寫入到新的Excel文件中，并保存到指定位置。

效果示例

本例為了簡化操作，總共有10個需要匯總的文件，如下所示：

每一個文件采用相同模板，格式一致，如下所示：

匯總后的文件，如下所示：

核心代碼

引入相關(guān)的模塊，如下所示：

import xlrd
import xlwt
import os

讀取excel內(nèi)容，如下所示：

def read_excel(excel_folder: str):
    """
    讀取Excel文檔
    :param excel_folder:
    :return:
    """
    excel_files = []  # excel文件列表
    excel_contents = []  # excel文件內(nèi)容
    files = os.listdir(path=excel_folder)
    if len(files) > 0:
        for file in files:
            if file.endswith('.xls'):
                excel_files.append(file)
    if len(excel_files) == 0:
        print('此目錄下沒有對應(yīng)的Excel文件')
    # 循環(huán)讀取文檔
    for file in excel_files:
        full_path = os.path.join(excel_folder, file)
        book = xlrd.open_workbook(filename=full_path)
        sheet = book.sheet_by_index(sheetx=0)
        # 取固定位置的值
        content = {
            '名稱': sheet.cell_value(1, 0),
            '年紀': sheet.cell_value(1, 1),
            '性別': sheet.cell_value(1, 2),
            '籍貫': sheet.cell_value(1, 3)
        }
        excel_contents.append(content)
 
return excel_contents

生成Excel內(nèi)容，如下所示：

def write_excel(excel_path: str, excel_contents: list):
    """
    生成Excel文檔
    :param excel_path:
    :return:
    """
    if len(excel_contents) == 0:
        print('沒有需要保存的內(nèi)容')
        return
    book = xlwt.Workbook(encoding='utf-8')
    sheet1 = book.add_sheet('統(tǒng)計信息')
    # 先寫標題
    sheet1.write(0, 0, '名稱')
    sheet1.write(0, 1, '年紀')
    sheet1.write(0, 2, '性別')
    sheet1.write(0, 3, '籍貫')
    i = 1
    for content in excel_contents:
        sheet1.write(i, 0, content['名稱'])
        sheet1.write(i, 1, content['年紀'])
        sheet1.write(i, 2, content['性別'])
        sheet1.write(i, 3, content['籍貫'])
        i += 1
    book.save(excel_path)

整體調(diào)用

excel_folder = r'D:\MyProject\Demo01\TESTEXCEL'
excel_path = r'D:\MyProject\Demo01\TESTEXCEL\匯總信息.xls'
excel_contents = read_excel(excel_folder)
# print(excel_contents)
write_excel(excel_path, excel_contents)
print('done')

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: