Files
doc_processer/create_table_pandas.py
2026-02-09 22:18:30 +08:00

92 lines
2.7 KiB
Python

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")