import sys

sys.path.insert(0, "/opt/homebrew/lib/python3.13/site-packages")

import pandas as pd

# Define 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]]

# Pad all lists to the same length
max_len = max(len(data_电动), len(data_电工), len(data_黑板), len(data_小), len(data_大))


def pad_list(lst, length):
    return lst + [["", ""]] * (length - len(lst))


data_电动 = pad_list(data_电动, max_len)
data_电工 = pad_list(data_电工, max_len)
data_黑板 = pad_list(data_黑板, max_len)
data_小 = pad_list(data_小, max_len)
data_大 = pad_list(data_大, max_len)

# Create DataFrame
df = pd.DataFrame(
    {
        "类别：电动": [row[0] for row in data_电动],
        "结果_1": [row[1] for row in data_电动],
        "类别：电工": [row[0] for row in data_电工],
        "结果_2": [row[1] for row in data_电工],
        "类别：黑板": [row[0] for row in data_黑板],
        "结果_3": [row[1] for row in data_黑板],
        "类别：小": [row[0] for row in data_小],
        "结果_4": [row[1] for row in data_小],
        "类别：大": [row[0] for row in data_大],
        "结果_5": [row[1] for row in data_大],
    }
)

# Replace empty strings with NaN for cleaner Excel output
df = df.replace("", pd.NA)

# Save to Excel
df.to_excel("数据表.xlsx", index=False, sheet_name="数据表")
print("Excel文件已创建: 数据表.xlsx")