Matplotlib自定义后端：非GUI环境下plt.show自动保存临时PNG

现有一段matplotlib代码在非GUI环境无法运行，需求是调用plt.show()时自动将图形保存为临时PNG文件：

import matplotlib.pyplot as plt
plt.scatter(2,3)
plt.scatter(4,5)
plt.show()

已知可以用plt.savefig替代plt.show()解决，但不愿修改原程序（可能由熟悉GUI环境的其他用户提供），故考虑通过修改matplotlibrc配置自定义后端以改变plt.show的行为。官方文档仅说明如何选择内置后端，提到可指定module://my_backend，但未定义自定义后端的接口（需实现哪些类/对象、命名规则），现询问是否有说明后端接口的文档，或其他修改plt.show行为的替代方案。

一、自定义后端的接口规范与实现 #

其实matplotlib的后端有明确的接口规范，官方虽然没把相关文档放在最显眼的位置，但你可以通过两种方式快速了解：

• 直接查看matplotlib内置后端的源码（比如backend_agg.py、backend_svg.py），这些是最直观的参考实例；
• 官方有专门的「后端开发指南」，里面详细说明了需要实现的类和方法细节。

简单来说，自定义后端需要覆盖以下核心部分：

1. 一个继承自matplotlib.backend_bases.FigureCanvasBase的画布类，至少要实现draw()方法（可以直接复用现有后端的渲染逻辑，比如AGG的FigureCanvasAgg）；
2. 一个FigureManager类（如果只是替换show()行为，简单继承FigureManagerBase即可）；
3. 必须暴露show()函数，这是plt.show()会直接调用的入口。

这里给你写一个极简的自定义后端示例（保存为my_backend.py）：

# my_backend.py
import tempfile
import matplotlib
from matplotlib.backends.backend_agg import FigureCanvasAgg
from matplotlib.backend_bases import FigureManagerBase

class FigureCanvasSaveTemp(FigureCanvasAgg):
    def show(self):
        # 创建临时PNG文件，delete=False保留文件以便后续访问
        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp_file:
            self.figure.savefig(tmp_file)
            print(f"图形已保存到临时文件: {tmp_file.name}")

class FigureManagerSaveTemp(FigureManagerBase):
    pass

def show():
    # 遍历所有打开的图形，调用各自画布的show方法
    for fig_manager in matplotlib._pylab_helpers.Gcf.get_all_fig_managers():
        fig_manager.canvas.show()

# 必须暴露的后端版本标识
backend_version = matplotlib.__version__

然后修改你的matplotlibrc配置文件，把backend项改成：

backend : module://my_backend

这样运行原代码时，plt.show()就会自动触发临时文件保存，完全不需要改动原程序。

二、更轻量的替代方案：猴子补丁 #

如果觉得自定义后端太繁琐，用「猴子补丁」直接替换plt.show()的实现会更简单，而且不需要修改配置文件。

在运行原代码之前，先执行这段补丁代码：

import tempfile
import matplotlib.pyplot as plt

# 保存原show方法（可选，如果你需要在某些场景下恢复默认行为）
original_show = plt.show

def patched_show(*args, **kwargs):
    # 遍历所有已创建的图形
    for fig_num in plt.get_fignums():
        fig = plt.figure(fig_num)
        # 创建临时文件并保存图形
        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp_file:
            fig.savefig(tmp_file)
            print(f"临时PNG文件路径: {tmp_file.name}")
    # 非GUI环境下原show方法会报错，所以可以注释掉下面一行
    # original_show(*args, **kwargs)

# 替换plt.show为自定义实现
plt.show = patched_show

之后再运行原代码，plt.show()就会自动把图形保存成临时PNG，完全不影响原代码的结构。

另外还有一种更简单的方式：先指定非GUI后端（比如AGG），再替换show函数：

import matplotlib
# 先设置非GUI后端，避免原show方法报错
matplotlib.use('Agg')
import tempfile
import matplotlib.pyplot as plt

def custom_show():
    for fig_num in plt.get_fignums():
        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp:
            plt.figure(fig_num).savefig(tmp)
            print(f"临时文件已生成: {tmp.name}")

plt.show = custom_show

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