如何用for循环优化Matplotlib直方图代码以消除重复逻辑？

当然可以通过for循环彻底消除重复逻辑，核心思路是提取所有重复代码中的可变元素（比如数据组、子图位置、标题、图例标签等），将它们整理成可遍历的集合，再通过循环批量完成绘图操作。

重构思路与示例代码 #

假设你原本的代码是四次重复调用plt.subplot+直方图绘制，重构后可以这样写：

import matplotlib.pyplot as plt
import numpy as np

# 模拟四组测试数据（替换为你的真实数据）
data_groups = [
    (np.random.randn(10)*2 + 10, np.random.randn(90)*2 + 5, "Group A"),
    (np.random.randn(10)*1.5 + 8, np.random.randn(90)*1.5 + 3, "Group B"),
    (np.random.randn(10)*3 + 12, np.random.randn(90)*3 + 6, "Group C"),
    (np.random.randn(10)*2.5 + 9, np.random.randn(90)*2.5 + 4, "Group D"),
]

# 创建2x2子图布局
fig, axes = plt.subplots(2, 2, figsize=(12, 10))
axes = axes.flatten()  # 将二维子图数组转为一维，方便循环遍历

# 循环处理每组数据
for ax, (top10_data, bottom90_data, group_name) in zip(axes, data_groups):
    # 绘制Top10和Bottom90直方图
    ax.hist(top10_data, bins=10, alpha=0.7, label="Top 10", color="#1f77b4")
    ax.hist(bottom90_data, bins=10, alpha=0.5, label="Bottom 90", color="#ff7f0e")
    
    # 设置子图属性
    ax.set_title(f"{group_name}: Top 10 vs Bottom 90")
    ax.set_xlabel("Value")
    ax.set_ylabel("Frequency")
    ax.legend()
    ax.grid(axis='y', linestyle='--', alpha=0.7)

plt.tight_layout()  # 自动调整子图间距
plt.show()

关键说明 #

• 提取可变元素：把四组数据、分组名称打包成元组列表data_groups，避免重复定义变量和绘图代码。
• 子图扁平化：用axes.flatten()把2x2的子图数组转为一维，直接和数据组一一对应遍历。
• 统一绘图逻辑：所有直方图的绘制、标签、标题设置都在循环内完成，只修改每组的专属参数，完全复用核心绘图代码。

这样重构后，代码量大幅减少，后续要新增或修改分组时，只需要调整data_groups即可，无需重复编写绘图逻辑。

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