将文件名分割字段生成xlsx文件

def excel_generate():
    # 获取当前目录下的所有文件
    files = os.listdir("D:/AutoTest/PythonProject/img")
    
    # 创建一个新的Excel工作簿
    workbook = Workbook()
    
    # 创建一个工作表
    worksheet = workbook.active
    
    # 添加表头
    worksheet.append(['名称', '指令类型', '执行次数'])
    
    # 遍历文件名并解析生成Excel数据
    for file in files:
        if file.endswith('.png'):
            # 解析文件名并根据下划线分割成三个部分
            parts = file[:-4].split('_')
            if len(parts) == 3:
                name, command_type, execution_count = parts
                worksheet.append(["D:\\AutoTest\\img\\"+file, int(command_type), int(execution_count)])

    # 保存Excel文件
    workbook.save('output.xlsx')
excel_generate()