Matplotlib 2.1.0导航栏Home按钮嵌入PyQt5窗口后功能异常

我之前踩过一模一样的坑——升级到Matplotlib 2.1.0之后，嵌入PyQt5窗口里的导航工具栏Home按钮不再回到第一次绘图后的状态，反而跳回初始的空白坐标轴。这是因为Matplotlib在2.1.0版本调整了导航栏Home按钮的默认逻辑，下面是具体的原因和解决办法：

问题根源 #

Matplotlib 2.1.0之前，导航栏的Home按钮会自动把第一次绘图后的坐标轴范围作为“Home”状态保存；但升级后，Home按钮默认绑定的是Axes.reset()方法，这个方法会直接重置坐标轴到创建时的空白初始状态，完全忽略你后续绘图的内容。

解决方案：自定义导航工具栏+手动保存Home状态 #

我们可以通过两个步骤修复这个问题：

1. 自定义一个导航工具栏类，重写Home按钮的行为
2. 在第一次绘图完成后，手动保存当前坐标轴的范围作为Home状态

1. 修改Matplotlib部件类文件（比如mpl_widget.py） #

首先创建自定义的导航工具栏，然后在绘图部件中使用它，并保存绘图后的坐标轴范围：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
from PyQt5.QtWidgets import QWidget, QVBoxLayout

# 自定义导航工具栏，重写Home按钮逻辑
class CustomNavigationToolbar(NavigationToolbar):
    def home(self, *args):
        # 优先使用我们手动保存的Home范围
        ax = self.canvas.figure.axes[0]
        if hasattr(ax, '_home_xlim') and hasattr(ax, '_home_ylim'):
            ax.set_xlim(ax._home_xlim)
            ax.set_ylim(ax._home_ylim)
            self.canvas.draw()
        else:
            # 如果没有保存的范围，调用默认行为
            super().home(*args)

# Matplotlib嵌入部件类
class MplWidget(QWidget):
    def __init__(self, parent=None):
        super(MplWidget, self).__init__(parent)
        self.figure = plt.figure()
        self.ax = self.figure.add_subplot(111)
        self.canvas = FigureCanvas(self.figure)
        # 使用自定义的导航工具栏
        self.toolbar = CustomNavigationToolbar(self.canvas, self)
        
        layout = QVBoxLayout()
        layout.addWidget(self.toolbar)
        layout.addWidget(self.canvas)
        self.setLayout(layout)
    
    def plot_data(self):
        # 清空坐标轴并绘图（替换成你的实际绘图逻辑）
        self.ax.clear()
        x = np.linspace(0, 10, 100)
        y = np.sin(x)
        self.ax.plot(x, y)
        self.ax.set_title("示例曲线")
        
        # 保存当前坐标轴范围作为Home状态
        self.ax._home_xlim = self.ax.get_xlim()
        self.ax._home_ylim = self.ax.get_ylim()
        
        # 更新画布
        self.canvas.draw()

2. 窗口文件（比如main_window.py，QtDesigner生成的文件） #

假设你的窗口文件里有一个plot_button按钮和一个mpl_widget实例，只需要给按钮绑定绘图方法即可：

from PyQt5.QtWidgets import QMainWindow
from ui_mainwindow import Ui_MainWindow  # 替换成你QtDesigner生成的UI类
from mpl_widget import MplWidget

class MainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        # 绑定绘图按钮点击事件
        self.plot_button.clicked.connect(self.mpl_widget.plot_data)

3. 主调用文件（比如main.py） #

这个文件不需要修改，保持原来的调用逻辑即可：

import sys
from PyQt5.QtWidgets import QApplication
from main_window import MainWindow

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

验证效果 #

按照你的操作步骤测试：

1. 点击绘图按钮前：显示空白坐标轴
2. 点击绘图按钮后：显示绘制的曲线，此时已经保存了当前坐标轴范围
3. 点击缩放按钮调整视图
4. 点击Home按钮：会回到步骤2的曲线视图，而不是初始空白坐标轴

这样就完美解决了Matplotlib 2.1.0升级后的Home按钮异常问题。

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