python实现自动化之文件合并

import xlrd
import xlwt
import os

def read_excel(excel_folder: str):
    """
    读取Excel文档
    :param excel_folder:
    :return:
    """
    excel_files = []  # excel文件列表
    excel_contents = []  # excel文件内容
    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('此目录下没有对应的Excel文件')
    # 循环读取文档
    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

def write_excel(excel_path: str, excel_contents: list):
    """
    生成Excel文档
    :param excel_path:
    :return:
    """
    if len(excel_contents) == 0:
        print('没有需要保存的内容')
        return
    book = xlwt.Workbook(encoding='utf-8')
    sheet1 = book.add_sheet('统计信息')
    # 先写标题
    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)

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')