循环中DataFrame、图形及输出文件的命名问题解决

核心问题修正 #

你之前用set存NAME_jour是关键错误——集合是无序的，而且不支持下标访问，直接改成列表就能解决TypeError，同时保证名称和DataFrame的顺序严格对应。

完整实现步骤 #

1. 准备匹配的有序列表 #

确保df_list（你的DataFrame列表）和NAME_jour（名称列表）是长度相同、顺序一一对应的列表：

# 替换成你的实际DataFrame和名称
df_list = [df_mon, df_tue, df_wed]
NAME_jour = ["周一客流数据", "周二客流数据", "周三客流数据"]

2. 用enumerate+zip循环关联索引、名称和DataFrame #

直接遍历三者的组合，彻底避免嵌套循环的重复命名和KeyError问题：

import seaborn as sns
import matplotlib.pyplot as plt

for idx, (name, df) in enumerate(zip(NAME_jour, df_list)):
    # 1. 执行你的DataFrame计算逻辑（示例为相关系数矩阵）
    calc_df = df.corr()
    
    # 2. 绘制热力图并设置对应标题
    plt.figure(figsize=(10, 8))
    sns.heatmap(calc_df, annot=True, cmap="YlGnBu")
    plt.title(f"客流热力图：{name}")
    
    # 3. 保存文件，文件名与标题一一对应
    # 可选：用名称命名（注意处理特殊字符）
    clean_name = name.replace(" ", "_").replace("/", "-")
    plt.savefig(f"heatmap_{clean_name}.png")
    # 或者用索引命名（更稳妥，避免名称异常）
    # plt.savefig(f"heatmap_{idx+1}_{name}.png")
    
    plt.close()  # 必须关闭画布，避免内存泄漏和图重叠

3. 常见问题规避 #

• 杜绝嵌套循环：之前的嵌套逻辑会重复遍历名称列表，导致同一名称被多次使用，触发命名冲突或KeyError，用zip直接绑定名称和DataFrame就不会出现这个问题。
• 名称合法性处理：如果名称包含空格、斜杠等特殊字符，先做清洗再用于文件名，避免保存失败。
• 画布清理：循环内必须调用plt.close()，否则会累积大量未关闭的画布，占用内存甚至导致后续图表样式混乱。

为什么之前的方法失效？ #

• 嵌套循环：重复遍历名称列表会导致同一名称被多次绑定不同的DataFrame，既会出现命名重复，也可能因DataFrame索引不匹配触发KeyError。
• 集合类型：set是无序且去重的，既不能通过下标访问元素，还会自动删除重复名称，导致名称列表和DataFrame列表长度不匹配，触发TypeError。

内容的提问来源于stack exchange，提问作者sam