如何无需迭代直接调用字典中同类对象的所有方法？（matplotlib场景）

嘿，这个需求太贴合实际场景了！经常需要同时操作多个figure或axes，还要避免写重复的循环代码对吧？Matplotlib本身没有内置这种批量调用的工具，但我们可以自己实现一个轻量的包装类，让你像调用单个对象方法一样，直接对字典里的所有同类对象执行操作。

方法一：自定义批量操作包装类 #

这个思路的核心是利用Python的__getattr__魔法方法，当你访问包装类的某个方法时，自动把这个方法分发到字典里的每一个对象上：

import matplotlib.pyplot as plt

class BatchPlotObjects:
    def __init__(self, obj_dict):
        # 存储传入的对象字典
        self.objects = obj_dict
    
    def __getattr__(self, method_name):
        # 当访问不存在的属性时，生成一个批量调用的函数
        def batch_executor(*args, **kwargs):
            # 遍历所有对象，调用指定方法，返回结果字典
            return {
                obj_name: getattr(obj, method_name)(*args, **kwargs)
                for obj_name, obj in self.objects.items()
            }
        return batch_executor

实际使用示例 #

1. 批量操作多个Figure

# 创建3个figure，存入字典
figs = {
    'scatter_plot': plt.figure(),
    'line_plot': plt.figure(),
    'bar_plot': plt.figure()
}

# 用包装类包裹字典
batch_figs = BatchPlotObjects(figs)

# 直接调用方法，无需迭代！
# 比如批量关闭所有figure
batch_figs.close()

# 再比如重新创建后批量设置标题
figs = {
    'scatter_plot': plt.figure(),
    'line_plot': plt.figure(),
    'bar_plot': plt.figure()
}
batch_figs = BatchPlotObjects(figs)
batch_figs.suptitle('Batch Generated Plot')

2. 批量操作多个Axes

# 创建子图并存入字典
fig, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2, 2)
axes = {
    'top_left': ax1,
    'top_right': ax2,
    'bottom_left': ax3,
    'bottom_right': ax4
}

batch_axes = BatchPlotObjects(axes)

# 批量设置坐标轴标签
batch_axes.set_xlabel('X Axis')
batch_axes.set_ylabel('Y Axis')

# 批量绘制曲线
import numpy as np
x = np.linspace(0, 2*np.pi, 100)
batch_axes.plot(x, np.sin(x), label='sin(x)')
batch_axes.legend()

为什么这个方案好用？ #

• 完全符合你的需求：不需要写for循环迭代字典，直接像调用单个对象方法一样操作批量对象
• 灵活通用：不管是Figure、Axes还是其他Matplotlib对象（比如Colorbar），只要是同类方法都能批量调用
• 保留返回值：批量调用后会返回一个字典，键是对象的名字，值是对应方法的返回结果，方便后续处理

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