如何修改Matplotlib Qt GUI中savefig的默认文件名？

我来帮你搞定这个问题——你说得没错，Matplotlib确实通过matplotlib.rcParams["savefig.directory"]控制保存对话框的默认目录，但默认文件名"image"是硬编码在工具栏逻辑里的，没法直接通过配置项修改。不过我们可以通过重写NavigationToolbar2QT的save_figure方法来实现自定义默认文件名（比如用你的self.currentFigure变量），具体操作如下：

1. 自定义导航工具栏类 #

创建一个继承自NavigationToolbar2QT的子类，重写save_figure方法，在调用Qt文件对话框时传入我们想要的默认文件名：

import os
import matplotlib
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT
from PyQt5 import QtWidgets

class CustomNavToolbar(NavigationToolbar2QT):
    def __init__(self, canvas, parent=None):
        super().__init__(canvas, parent)
        # 这里替换成你实际的self.currentFigure变量，示例假设它在主窗口类中
        self.default_save_name = self.parent().currentFigure

    def save_figure(self, *args):
        # 读取rcParams里配置的默认保存目录
        save_dir = matplotlib.rcParams["savefig.directory"]
        # 拼接默认路径：目录 + 自定义文件名
        default_path = os.path.join(save_dir, self.default_save_name)
        
        # 弹出Qt的保存对话框，指定默认文件名
        file_path, _ = QtWidgets.QFileDialog.getSaveFileName(
            self.parent(),
            "Save Your Figure",
            default_path,
            # 保持和Matplotlib默认一致的文件格式过滤器
            "PNG (*.png);;JPEG (*.jpg *.jpeg);;PDF (*.pdf);;SVG (*.svg);;EPS (*.eps)"
        )
        
        if file_path:
            # 如果用户选择了路径，调用Matplotlib的保存方法
            self.canvas.figure.savefig(file_path)

2. 在你的Qt应用中替换默认工具栏 #

把原来使用的NavigationToolbar2QT换成我们自定义的CustomNavToolbar就行：

# 假设你已经有了canvas对象和主窗口实例main_window
toolbar = CustomNavToolbar(canvas, main_window)
main_window.addToolBar(toolbar)

关键细节说明 #

• 为什么要重写？因为Matplotlib默认的save_figure方法里，默认文件名是固定写死的"image"，没有提供对应的rcParams配置项，所以只能通过重写这个方法来替换默认值。
• 你可以动态更新default_save_name的值，比如在切换图表时，把它设置成当前图表的名称，这样每次打开保存对话框时默认文件名都会自动跟着变化。
• 代码里保留了从savefig.directory读取默认目录的逻辑，和你原来的需求完全兼容。

灵活调整的小技巧 #

如果你的currentFigure不是在主窗口里，而是在其他地方，也可以在初始化自定义工具栏时直接传进去：

class CustomNavToolbar(NavigationToolbar2QT):
    def __init__(self, canvas, parent=None, default_name="image"):
        super().__init__(canvas, parent)
        self.default_save_name = default_name

    # save_figure方法和上面一致...

# 使用时直接传入你的currentFigure
toolbar = CustomNavToolbar(canvas, main_window, default_name=self.currentFigure)

这样就能更灵活地传递自定义的默认文件名了。

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