Python Pandas 自动验单数据处理 - 从 CSV 文件中提取并合并到 Excel 文件

import pandas as pd
import os
import re
from openpyxl import load_workbook

output_folder = 'F:/自动验单数据/验本数据/'
file_format = r'.*.csv'
folder_path = 'F:/自动验单数据/原始数据/'

for filename in os.listdir(folder_path):
    if re.match(file_format, filename):
        date_str = filename[11:19]
        file_path = os.path.join(folder_path, filename)
        yb_data = pd.read_csv(file_path, dtype=object, usecols=['发验机构', '上下行', '发验时间'])
        data = yb_data[((yb_data['发验机构'] == '21506') & (yb_data['地市'] == '扬州市') & (yb_data['上下行'] == '上行'))]
        file_name = date_str + '验本数据.xlsx'
        output_file = os.path.join(output_folder, file_name)
        
        if os.path.exists(output_file):
            # 如果文件已存在，则追加数据到现有工作簿
            wb = load_workbook(output_file)
            ws = wb.active
            data.to_excel(output_file, index=False, header=False, startrow=ws.max_row, engine='openpyxl')
            wb.save(output_file)
        else:
            # 如果文件不存在，则创建新工作簿并写入数据
            data.to_excel(output_file, index=False, engine='openpyxl')

print('匹配完成！')