如何设置Matplotlib画布内置保存按钮的默认路径？

Matplotlib的内置保存按钮默认路径由使用的GUI后端（如QtAgg、TkAgg）决定，没有直接的全局配置参数，但可以通过修改后端文件对话框的初始目录实现需求。以下是针对常见后端的实现方法：

方法1：针对QtAgg后端（推荐，Matplotlib默认后端之一） #

通过获取Qt窗口的保存动作，重写对话框的初始路径：

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
from PyQt5.QtWidgets import QFileDialog

# 确认使用QtAgg后端（若当前不是可手动设置）
mpl.use('QtAgg')

fig, ax = plt.subplots()
ax.imshow(np.random.random((10,10)))

# 获取画布对应的Qt窗口
canvas = fig.canvas
window = canvas.parent()

# 从工具栏找到保存按钮对应的动作
toolbar = canvas.toolbar
save_action = None
for action in toolbar.actions():
    if action.toolTip() == 'Save the figure':
        save_action = action
        break

# 重写保存动作的触发逻辑
def custom_save():
    # 指定默认路径，需确保路径已存在
    default_path = "./saved_plots/"
    # 弹出自定义初始目录的保存对话框
    file_path, _ = QFileDialog.getSaveFileName(
        window, "Save Figure", default_path, 
        "Image Files (*.png *.jpg *.pdf *.svg)"
    )
    if file_path:
        fig.savefig(file_path)

# 替换原保存动作的触发事件
if save_action:
    save_action.triggered.disconnect()
    save_action.triggered.connect(custom_save)

plt.show()

方法2：针对TkAgg后端 #

如果使用TkAgg后端，可通过修改Tk文件对话框的初始目录：

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
import tkinter as tk
from tkinter import filedialog

mpl.use('TkAgg')

fig, ax = plt.subplots()
ax.imshow(np.random.random((10,10)))

canvas = fig.canvas
root = canvas.get_tk_widget().master

# 获取工具栏的保存按钮
toolbar = canvas.toolbar
save_button = None
for child in toolbar.winfo_children():
    if child.cget('text') == 'Save':
        save_button = child
        break

# 重写保存按钮的命令逻辑
def custom_save():
    default_path = "./saved_plots/"
    # 设置对话框初始目录
    file_path = filedialog.asksaveasfilename(
        initialdir=default_path,
        defaultextension=".png",
        filetypes=[("Image Files", "*.png *.jpg *.pdf *.svg"), ("All Files", "*.*")]
    )
    if file_path:
        fig.savefig(file_path)

if save_button:
    save_button.config(command=custom_save)

plt.show()

注意事项 #

• 先通过mpl.get_backend()查看当前使用的后端，选择对应方法
• 确保指定的默认路径已创建，否则对话框会跳转到系统默认目录
• 不同Matplotlib版本的工具栏按钮提示文本可能略有差异，若找不到动作可调整判断条件（比如检查tooltip文本）

你之前尝试的按键绑定是自定义保存逻辑，而上述代码直接替换了内置保存按钮的触发逻辑，实现默认路径的设置。

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