from openpyxl import Workbook
from openpyxl.styles import Font, Alignment, Border, Side

wb = Workbook()
sheet = wb.active
sheet.title = "数据表"

# Headers
headers = ["类别：电动", "结果", "类别：电工", "结果", "类别：黑板", "结果", "类别：小", "结果", "类别：大", "结果"]

for col, header in enumerate(headers, start=1):
    cell = sheet.cell(row=1, column=col, value=header)
    cell.font = Font(bold=True)
    cell.alignment = Alignment(horizontal="center")

# Data for each category
data_电动 = [
    ["2615 - 243", 2372],
    ["2633 - 244", 2389],
    ["2542 - 243", 2299],
    ["2453 - 369", 2084],
    ["2670 - 244", 2426],
    ["2416 - 357", 2059],
    ["2574 - 239", 2335],
    ["2641 - 243", 2398],
    ["2640 - 242", 2398],
    ["2257 - 244", 2013],
    ["3114 - 410", 2704],
    ["2815 - 412", 2403],
    ["2545 - 243", 2302],
    ["2299 - 243", 2056],
]

data_电工 = [["2443 - 20", 2423], ["2460 - 13", 2447], ["2339 - 17", 2322], ["2405 - 13", 2392], ["2661 - 16", 2645], ["3065 - 15", 3050], ["2912 - 14", 2898], ["2883 - 14", 2869]]

data_黑板 = [
    ["1902 - 14", 1888],
    ["2319 - 14", 2305],
    ["2041 - 16", 2025],
    ["2451 - 13", 2438],
    ["1993 - 14", 1979],
    ["1841 - 18", 1823],
    ["2083 - 16", 2067],
    ["2009 - 18", 1991],
    ["2001 - 16", 1985],
    ["2014 - 14", 2000],
    ["2073 - 17", 2056],
    ["2008 - 15", 1993],
    ["2030 - 16", 2014],
    ["1928 - 22", 1906],
    ["1991 - 19", 1972],
    ["2235 - 12", 2223],
    ["2335 - 19", 2316],
    ["1920 - 16", 1904],
    ["1942 - 16", 1926],
    ["1974 - 19", 1955],
    ["1863 - 16", 1847],
    ["512 - 15", 497],
]

data_小 = [["1881 - 20", 1861], ["2055 - 22", 2033], ["2034 - 20", 2014], ["1981 - 17", 1964], ["1629 - 12", 1617], ["913 - 18", 895], ["1842 - 19", 1823], ["1891 - 20", 1871]]

data_大 = [["1931 - 20", 1911], ["1775 - 11", 1764], ["1885 - 21", 1864], ["120 - 18", 102], ["1675 - 13", 1662]]

# Write data starting from row 2
row = 2
max_rows = max(len(data_电动), len(data_电工), len(data_黑板), len(data_小), len(data_大))

for i in range(max_rows):
    # 电动
    if i < len(data_电动):
        sheet.cell(row=row + i, column=1, value=data_电动[i][0])
        sheet.cell(row=row + i, column=2, value=data_电动[i][1])

    # 电工
    if i < len(data_电工):
        sheet.cell(row=row + i, column=3, value=data_电工[i][0])
        sheet.cell(row=row + i, column=4, value=data_电工[i][1])

    # 黑板
    if i < len(data_黑板):
        sheet.cell(row=row + i, column=5, value=data_黑板[i][0])
        sheet.cell(row=row + i, column=6, value=data_黑板[i][1])

    # 小
    if i < len(data_小):
        sheet.cell(row=row + i, column=7, value=data_小[i][0])
        sheet.cell(row=row + i, column=8, value=data_小[i][1])

    # 大
    if i < len(data_大):
        sheet.cell(row=row + i, column=9, value=data_大[i][0])
        sheet.cell(row=row + i, column=10, value=data_大[i][1])

# Set column widths
for col in range(1, 11):
    sheet.column_dimensions[chr(64 + col)].width = 15

wb.save("数据表.xlsx")
print("Excel文件已创建: 数据表.xlsx")