matplotlib + PyQt5：如何实现类间参数传递？

我经常处理Qt与Matplotlib结合的嵌入式绘图场景，这里有几个实用且符合Qt开发习惯的方案，帮你实现参数传递：

方法1：初始化时直接传入参数 #

这是最直观的方式，在创建MyStaticMplCanvas实例的同时，把需要的参数一起传递过去，画布类在构造函数中接收并存储，用于后续绘图逻辑。

修改ApplicationWindow代码： #

class ApplicationWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        self.setupUi()
        
        # 示例参数：可以是你需要传递的任何数据（如数组、配置项等）
        self.plot_x = [1, 2, 3, 4, 5]
        self.plot_y = [2, 4, 6, 8, 10]
        
        # 创建画布时传入参数
        self.canvas = MyStaticMplCanvas(
            self, 
            width=5, 
            height=4, 
            dpi=100, 
            x_data=self.plot_x, 
            y_data=self.plot_y
        )
        
        # 后续布局添加等操作保持不变...

修改MyStaticMplCanvas代码： #

class MyStaticMplCanvas(FigureCanvas):
    def __init__(self, parent=None, width=5, height=4, dpi=100, x_data=None, y_data=None):
        # 初始化Matplotlib Figure和Axes
        fig = Figure(figsize=(width, height), dpi=dpi)
        self.axes = fig.add_subplot(111)
        super().__init__(fig)
        self.setParent(parent)
        
        # 保存传递的参数（设置默认值避免空指针）
        self.x_data = x_data or []
        self.y_data = y_data or []
        
        # 调用绘图方法
        self._plot_data()
    
    def _plot_data(self):
        # 使用接收的参数绘制图形
        self.axes.clear()
        self.axes.plot(self.x_data, self.y_data, marker='o')
        self.draw()

方法2：用Setter方法动态更新参数 #

如果参数需要在程序运行过程中动态修改（比如用户输入新数据后更新图表），这种方法更灵活。先创建画布实例，之后通过专门的方法传递参数并触发重绘。

修改MyStaticMplCanvas代码： #

class MyStaticMplCanvas(FigureCanvas):
    def __init__(self, parent=None, width=5, height=4, dpi=100):
        fig = Figure(figsize=(width, height), dpi=dpi)
        self.axes = fig.add_subplot(111)
        super().__init__(fig)
        self.setParent(parent)
        
        # 初始化默认参数
        self.x_data = []
        self.y_data = []
    
    def set_plot_parameters(self, x_data, y_data):
        # 更新内部参数
        self.x_data = x_data
        self.y_data = y_data
        # 触发重绘
        self._plot_data()
    
    def _plot_data(self):
        self.axes.clear()
        self.axes.plot(self.x_data, self.y_data)
        self.draw()

在ApplicationWindow中调用Setter： #

class ApplicationWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        self.setupUi()
        
        self.canvas = MyStaticMplCanvas(self)
        # 绑定按钮点击事件（示例：用户点击按钮后更新数据）
        self.update_btn.clicked.connect(self._update_canvas)
    
    def _update_canvas(self):
        # 模拟获取新参数（可以是从输入框读取、数据库查询等）
        new_x = [1, 3, 5, 7, 9]
        new_y = [1, 9, 25, 49, 81]
        
        # 调用Setter传递参数
        self.canvas.set_plot_parameters(new_x, new_y)

方法3：利用Qt信号槽机制传递参数 #

如果参数更新是由Qt事件触发的（比如菜单操作、定时器触发），使用信号槽是最符合Qt设计模式的方式，松耦合且易于维护。

修改ApplicationWindow代码： #

class ApplicationWindow(QtWidgets.QMainWindow):
    # 定义信号：参数类型根据你的需求调整（这里是两个列表）
    plot_data_updated = QtCore.pyqtSignal(list, list)
    
    def __init__(self):
        super().__init__()
        self.setupUi()
        
        self.canvas = MyStaticMplCanvas(self)
        # 连接信号与画布的槽函数
        self.plot_data_updated.connect(self.canvas.on_data_updated)
        
        # 示例：绑定菜单操作触发信号
        self.refresh_plot_action.triggered.connect(self._emit_new_data)
    
    def _emit_new_data(self):
        # 生成或获取新参数
        new_x = [2, 4, 6, 8, 10]
        new_y = [4, 16, 36, 64, 100]
        
        # 发射信号传递参数
        self.plot_data_updated.emit(new_x, new_y)

修改MyStaticMplCanvas代码： #

class MyStaticMplCanvas(FigureCanvas):
    def __init__(self, parent=None, width=5, height=4, dpi=100):
        fig = Figure(figsize=(width, height), dpi=dpi)
        self.axes = fig.add_subplot(111)
        super().__init__(fig)
        self.setParent(parent)
    
    @QtCore.pyqtSlot(list, list)
    def on_data_updated(self, x_data, y_data):
        # 接收信号传递的参数并更新图表
        self.axes.clear()
        self.axes.plot(x_data, y_data, color='r')
        self.draw()

你可以根据自己的场景选择最合适的方法：如果参数是固定的，用方法1；如果需要动态更新，方法2更简单；如果是Qt事件驱动的更新，方法3更优雅。

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