生成excel表格

import openpyxl

def generate_excel(data, columns, file_path):
    workbook = openpyxl.Workbook()
    sheet = workbook.active

    # 写入列名
    for col_idx, column in enumerate(columns, start=1):
        cell = sheet.cell(row=1, column=col_idx)
        cell.value = column

    # 写入数据
    for row_idx, row in enumerate(data, start=2):
        for col_idx, value in enumerate(row, start=1):
            cell = sheet.cell(row=row_idx, column=col_idx)
            cell.value = value

    # 保存Excel文件
    workbook.save(file_path)

# 示例数据
columns = ['Name', 'Age', 'Country']
data = [
    ['John', 25, 'USA'],
    ['Alice', 30, 'UK'],
    ['Bob', 35, 'Canada'],
]

# 生成Excel文件
generate_excel(data, columns, 'output.xlsx')