PyQt5中Matplotlib绘图无法更新的问题求助

开发一款包含函数绘图功能的GUI时，遇到绘图无法更新的问题。曾尝试让plotUpdate函数循环执行，结果程序直接崩溃。以下是简化后的代码：

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        # 添加按钮、输入框、标签等组件...

        self.fig = None
        self.ax = None
        self.x = np.arange(-10,10,0.01)
    
        self.plotButton.clicked.connect(self.run)
        self.updatePlotButton.clicked.connect(self.plotUpdate)

    def quad_cubic(self):
        c0value = float(self.c0.text())
        c1value = float(self.c1.text())
        c2value = float(self.c2.text())
        c3value = float(self.c3.text())
        y2 = c0value + c1value*self.x + c2value*self.x**2
        y3 = c0value + c1value*self.x + c2value*self.x**2 + c3value*self.x**3
        return y2,y3

    def run(self):
        self.fig,self.ax = plt.subplots(2,1,dpi = 150)
        self.plotUpdate()
    
    def plotUpdate(self):
        y2,y3 = self.quad_cubic()
        self.ax[0].cla()
        self.ax[1].cla()
        self.ax[0].plot(self.x,y2)
        self.ax[1].plot(self.x,y3)      
        plt.show(block = False)

问题分析 #

当前代码的核心问题：

• 直接用plt.subplots和plt.show创建独立matplotlib窗口，与PyQt的GUI事件循环冲突，导致更新失效甚至崩溃
• 每次更新都调用cla()清除轴再重新绘图，效率低且易引发渲染问题
• 未正确利用Qt后端的画布更新机制

修正方案 #

要实现稳定的绘图更新，需将matplotlib画布嵌入PyQt窗口，并通过更新已有线条数据的方式刷新绘图，而非重复创建绘图对象。

修改后的代码示例 #

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

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        # 原有的按钮、输入框、标签等组件初始化代码...
        
        # 创建matplotlib画布并嵌入GUI
        self.fig, self.ax = plt.subplots(2, 1, dpi=150)
        self.canvas = FigureCanvas(self.fig)
        self.toolbar = NavigationToolbar(self.canvas, MainWindow)
        
        # 将画布和工具栏添加到GUI布局（需对应你的窗口布局结构）
        layout = QVBoxLayout()
        layout.addWidget(self.toolbar)
        layout.addWidget(self.canvas)
        central_widget = QWidget()
        central_widget.setLayout(layout)
        MainWindow.setCentralWidget(central_widget)
        
        self.x = np.arange(-10, 10, 0.01)
        self.line2 = None  # 保存二次曲线线条对象
        self.line3 = None  # 保存三次曲线线条对象
        
        self.plotButton.clicked.connect(self.run)
        self.updatePlotButton.clicked.connect(self.plotUpdate)

    def quad_cubic(self):
        c0value = float(self.c0.text())
        c1value = float(self.c1.text())
        c2value = float(self.c2.text())
        c3value = float(self.c3.text())
        y2 = c0value + c1value*self.x + c2value*self.x**2
        y3 = c0value + c1value*self.x + c2value*self.x**2 + c3value*self.x**3
        return y2, y3

    def run(self):
        # 首次绘图，创建并保存线条对象
        y2, y3 = self.quad_cubic()
        self.line2, = self.ax[0].plot(self.x, y2)
        self.line3, = self.ax[1].plot(self.x, y3)
        self.canvas.draw()
    
    def plotUpdate(self):
        # 更新已有线条的数据，无需清除轴或重建绘图
        y2, y3 = self.quad_cubic()
        self.line2.set_ydata(y2)
        self.line3.set_ydata(y3)
        
        # 自动调整轴范围，确保新数据完整显示（可选）
        self.ax[0].relim()
        self.ax[0].autoscale_view()
        self.ax[1].relim()
        self.ax[1].autoscale_view()
        
        # 高效刷新画布
        self.canvas.draw_idle()

关键改动说明 #

• 使用FigureCanvasQTAgg将matplotlib画布嵌入PyQt窗口，统一事件循环，避免窗口冲突
• 首次绘图时保存线条对象，后续更新仅修改线条的ydata，提升效率
• 用canvas.draw_idle()替代plt.show()，实现高效的画布刷新
• 添加轴范围自动调整逻辑，确保新数据能完整展示（可根据需求选择是否保留）

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