如何在 PyQt5 中嵌入 Matplotlib 图形？

在 PyQt5 应用程序中嵌入 Matplotlib 图形可以提供丰富的数据可视化功能。本文将引导您通过几个简单的步骤来实现在 PyQt5 中嵌入 Matplotlib 图形的过程。

步骤一：安装必要的库

首先，确保您的环境已经安装了以下库：

• PyQt5：用于创建 PyQt5 应用程序界面的库
• Matplotlib：用于绘制图形的库 您可以使用以下命令在终端中安装这些库：

pip install pyqt5 matplotlib

步骤二：创建主窗口

在 PyQt5 中创建一个主窗口是实现图形嵌入的第一步。您可以使用以下代码创建一个简单的主窗口类：

from PyQt5.QtWidgets import QApplication, QMainWindow

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("Matplotlib in PyQt5")
        self.setGeometry(100, 100, 800, 600)

步骤三：创建 Matplotlib 子窗口

在主窗口中创建一个子窗口，用于显示 Matplotlib 图形。您可以使用 Matplotlib 的 FigureCanvas 类将 Matplotlib 图形嵌入到 PyQt5 应用程序中。

from PyQt5.QtWidgets import QWidget, QVBoxLayout
from matplotlib.backends.backend_qt5agg import FigureCanvas
from matplotlib.figure import Figure

class MatplotlibWidget(QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.figure = Figure()
        self.canvas = FigureCanvas(self.figure)

        layout = QVBoxLayout()
        layout.addWidget(self.canvas)
        self.setLayout(layout)

步骤四：绘制图形

在 MatplotlibWidget 类中添加绘制图形的方法。

from numpy import linspace

class MatplotlibWidget(QWidget):
    # ...

    def draw_graph(self):
        ax = self.figure.add_subplot(111)
        x = linspace(0, 10, 100)
        y = x ** 2
        ax.plot(x, y)
        self.canvas.draw()

步骤五：将 MatplotlibWidget 添加到主窗口

在主窗口类的 __init__ 方法中实例化 MatplotlibWidget，并将其添加到主窗口中。

class MainWindow(QMainWindow):
    def __init__(self):
        # ...
        self.matplotlib_widget = MatplotlibWidget(self)
        self.setCentralWidget(self.matplotlib_widget)

步骤六：运行应用程序

在 if __name__ == '__main__' 代码块中启动 PyQt5 应用程序。

if __name__ == '__main__':
    app = QApplication([])
    main_window = MainWindow()
    main_window.show()
    app.exec_()

现在，您的 PyQt5 应用程序中已经成功嵌入了 Matplotlib 图形，可以通过调用 MatplotlibWidget 的 draw_graph 方法绘制图形。

以上步骤提供了一个基本的框架，您可以根据应用程序的需求进行自定义，如设置图形样式、添加交互功能等。

希望本文对您在 PyQt5 中嵌入 Matplotlib 图形的过程有所帮助！