告别手动统计!用Python一键汇总Excel水果数量,小白也能轻松上手

import pandas as pdimport refrom openpyxl import Workbookfrom openpyxl.styles import Font, PatternFill, Alignment, Border, Sidefrom collections import defaultdictdf = pd.read_excel('data0215.xlsx')# ── 解析函数 ────────────────────────────────────────────────────def parse_name_qty(name_str, total_qty):    """    解析名称字符串和总数量，返回 {水果名: 数量} 字典。    - 单名称（如 '苹果'）→ 全部数量归该名称    - 多名称（如 '橙子1 柠檬2'）→ 按比例分配    """    name_str = str(name_str).strip()    # 匹配 "汉字+数字" 对    pairs = re.findall(r'([\u4e00-\u9fa5]+)(\d+)', name_str)    if not pairs:        # 纯名称，无内嵌比例数字        return {name_str: total_qty}    total_ratio = sum(int(r) for _, r in pairs)    result = {}    for name, ratio in pairs:        result[name] = total_qty * int(ratio) / total_ratio    return result# ── 汇总逻辑 ─────────────────────────────────────────────────────GROUP_COUNT = 3  # 名称/数量列的组数records = []for _, row in df.iterrows():    seq = row['序号']    totals = defaultdict(float)    for i in range(1, GROUP_COUNT + 1):        name_col = f'名称 {i}'        qty_col  = f'数量 {i}'        if name_col not in row or pd.isna(row[name_col]):            continue        dist = parse_name_qty(row[name_col], row[qty_col])        for fruit, qty in dist.items():            totals[fruit] += qty    record = {'序号': seq}    record.update(totals)    records.append(record)result_df = pd.DataFrame(records).fillna('')# 固定列顺序（按题意）fruits = ['苹果', '橙子', '柠檬', '香蕉']cols = ['序号'] + [f for f in fruits if f in result_df.columns]result_df = result_df[cols]# 整数化（有值的格子）for col in fruits:    if col in result_df.columns:        result_df[col] = result_df[col].apply(lambda x: int(x) if x != '' else '')print("汇总结果：")print(result_df.to_string(index=False))# ── 输出到 Excel ─────────────────────────────────────────────────out_path = '汇总结果.xlsx'wb = Workbook()ws = wb.activews.title = '汇总'header_fill = PatternFill('solid', start_color='4472C4')header_font = Font(bold=True, color='FFFFFF', name='Arial')cell_font   = Font(name='Arial', size=11)center      = Alignment(horizontal='center', vertical='center')thin        = Side(style='thin', color='BFBFBF')border      = Border(left=thin, right=thin, top=thin, bottom=thin)# 写表头for col_idx, col_name in enumerate(result_df.columns, 1):    cell = ws.cell(row=1, column=col_idx, value=col_name)    cell.fill   = header_fill    cell.font   = header_font    cell.alignment = center    cell.border = border# 写数据alt_fill = PatternFill('solid', start_color='EBF3FB')for row_idx, row in enumerate(result_df.itertuples(index=False), 2):    fill = alt_fill if row_idx % 2 == 0 else PatternFill()    for col_idx, value in enumerate(row, 1):        cell = ws.cell(row=row_idx, column=col_idx, value=value if value != '' else None)        cell.font      = cell_font        cell.alignment = center        cell.border    = border        cell.fill      = fill# 列宽ws.column_dimensions['A'].width = 8for col_letter in ['B', 'C', 'D', 'E']:    ws.column_dimensions[col_letter].width = 10ws.row_dimensions[1].height = 20wb.save(out_path)print(f"\n已保存至：{out_path}")