如何在循环中为不同Matplotlib Figure动态分配Subplot

我明白你的需求——循环生成子图时，不想让单张Figure挤太多子图，希望满额后自动新开一张，还不想用Python不推荐的动态变量命名，想用字典来管理Figure和Axes对吧？我来给你捋捋怎么实现：

核心思路 #

• 用字典存储所有生成的Figure对象，用自定义键（比如fig_1、fig_2）区分，避免动态变量的维护麻烦
• 提前设定每张Figure能容纳的最大子图数，每次添加子图前检查当前图的子图数量，满额就自动新建Figure
• 动态计算子图的行列布局，不用手动写固定的Position数组

修改后的完整可运行代码 #

import matplotlib.pyplot as plt
import numpy as np

# 配置参数
max_subplots_per_fig = 5  # 每张图最多放5个子图
cols = 2  # 子图列数
# 自动计算需要的行数（向上取整，确保能放下所有子图）
rows = (max_subplots_per_fig + cols - 1) // cols

# 用字典存储所有Figure对象，键为自定义标识
figs = {}
current_fig_index = 0  # 当前使用的Figure序号
current_subplot_count = 0  # 当前Figure已添加的子图数

def add_subplot(fig, subplot_pos, data_idx):
    """给指定Figure添加子图并完成绘图逻辑"""
    ax = fig.add_subplot(rows, cols, subplot_pos)
    ax.plot(range(1, 10), np.ones(9) * data_idx, label="first line")
    ax.plot(range(1, 10), np.ones(9) * 2*data_idx, label="second line")
    ax.set_title(f"Subplot {data_idx+1}")
    ax.set_xlabel("x label")
    return ax

# 模拟循环处理20个元素（替换成你的实际数据数量即可）
for i in range(20):
    # 触发条件：当前Figure不存在，或已填满子图 → 新建Figure
    if current_fig_index not in figs or current_subplot_count >= max_subplots_per_fig:
        current_fig_index += 1
        fig_key = f"fig_{current_fig_index}"
        figs[fig_key] = plt.figure(current_fig_index, figsize=(10, 7))
        current_subplot_count = 0  # 重置当前图的子图计数
    
    current_subplot_count += 1
    # 给当前Figure添加子图
    current_ax = add_subplot(figs[f"fig_{current_fig_index}"], current_subplot_count, i)
    
    # 给每张Figure添加公共图例（当前图填满，或处理到最后一个元素时触发）
    if current_subplot_count == max_subplots_per_fig or i == 19:
        handles, labels = current_ax.get_legend_handles_labels()
        figs[f"fig_{current_fig_index}"].legend(handles, labels, loc='lower right')

plt.show()

关键细节解释 #

• 字典管理Figure：figs字典通过自定义键来管理所有生成的Figure，后续要操作某张图时直接通过键访问，比动态变量更安全、更易维护
• 自动新建逻辑：每次循环先检查当前图的子图数是否达上限，满了就自动新建Figure并重置计数，完全不用手动提前创建多张图
• 行列动态计算：rows = (max_subplots_per_fig + cols - 1) // cols是Python中向上取整的常用写法，比如5个子图2列会自动算出3行，刚好适配布局
• 图例统一处理：在每张图的最后一个子图添加完成后，给整个Figure添加公共图例，不管是图填满了还是处理到最后一个元素，都能保证每张图都有图例

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