第9课:打通Python与Excel——预测结果自动回填报表,告别“复制粘贴手”

import openpyxlfrom openpyxl.styles import Font, PatternFill, Alignment, Border, Sidefrom openpyxl.utils import get_column_letter# ==========================================# 1. 数据准备# ==========================================customers = [    {"姓名": "张三", "消费金额": 8500, "标签": "高价值"},    {"姓名": "李四", "消费金额": 3200, "标签": "中价值"},    {"姓名": "王五", "消费金额": 800, "标签": "低价值"},    {"姓名": "赵六", "消费金额": 9200, "标签": "高价值"},    {"姓名": "孙七", "消费金额": 1500, "标签": "低价值"},]# ==========================================# 2. 创建Excel与样式定义# ==========================================wb = openpyxl.Workbook()ws = wb.activews.title = "客户价值分析表"# --- 定义配色方案 ---# 表头样式header_font = Font(bold=True, color="FFFFFF", size=12)header_fill = PatternFill(start_color="2F5496", fill_type="solid") # 以此深蓝为主色调# 边框样式 (让表格看起来更整齐)thin_border = Border(    left=Side(style='thin'),    right=Side(style='thin'),    top=Side(style='thin'),    bottom=Side(style='thin'))center_align = Alignment(horizontal="center", vertical="center")# 标签对应的颜色 (使用色阶，比纯色更高级)TAG_COLORS = {"高价值": {"fill": PatternFill(start_color="FCE4D6", fill_type="solid"), "font": Font(color="C00000", bold=True)},# 浅红底深红字"中价值": {"fill": PatternFill(start_color="FFEB9C", fill_type="solid"), "font": Font(color="9C6500", bold=True)},# 浅黄底深黄字"低价值": {"fill": PatternFill(start_color="C6EFCE", fill_type="solid"), "font": Font(color="006100", bold=True)}# 浅绿底深绿字}# ==========================================# 3. 写入数据# ==========================================# 写入表头ws.append(["客户姓名", "消费金额", "客户分层"])# 写入数据行total_amount = 0for cust in customers:    row_data = [cust["姓名"], cust["消费金额"], cust["标签"]]    ws.append(row_data)    total_amount += cust["消费金额"]# ==========================================# 4. 样式应用与美化# ==========================================# 获取数据行数 max_row = ws.max_rowmax_col = ws.max_column# A. 处理表头for cell in ws[1]: # 第一行    cell.font = header_font    cell.fill = header_fill    cell.border = thin_border    cell.alignment = center_align# B. 处理数据行for row in ws.iter_rows(min_row=2, max_row=max_row, min_col=1, max_col=max_col):# 获取当前行的标签 (第3列，索引2)    tag = row[2].value    style_config = TAG_COLORS.get(tag, TAG_COLORS["低价值"]) # 默认给低价值样式for cell in row:        cell.border = thin_border        cell.alignment = center_align# 应用填充色        cell.fill = style_config["fill"]# 如果是金额列，应用千分位格式 (虽然openpyxl处理格式较麻烦，这里仅做数值展示)if cell.column == 2:            cell.number_format = '#,##0'# 单独设置标签列的字体颜色    row[2].font = style_config["font"]# ==========================================# 5. 添加合计行 # ==========================================# 追加合计行ws.append(["总计", total_amount, "-"])total_row_idx = max_row + 1# 设置合计行样式ws[f"A{total_row_idx}"].font = Font(bold=True)ws[f"B{total_row_idx}"].font = Font(bold=True, size=12)ws[f"B{total_row_idx}"].fill = PatternFill(start_color="D9D9D9", fill_type="solid") # 灰色底# ==========================================# 6. 调整列宽# ==========================================column_widths = [15, 15, 15] # 基础宽度for i, width in enumerate(column_widths, 1):    col_letter = get_column_letter(i)    ws.column_dimensions[col_letter].width = width# ==========================================# 7. 保存# ==========================================wb.save("客户价值分析报告_美化版.xlsx")print("✅ 报表生成完毕！请查看 '客户价值分析报告_美化版.xlsx'")