将matplotlib嵌入pyside2

需要的库

from PySide2.QtWidgets import *
from PySide2.QtUiTools import QUiLoader
from PySide2.QtCore import QFile

from matplotlib.backends.backend_qt5agg import (
    FigureCanvasQTAgg as FigureCanvas, NavigationToolbar2QT as NavigationToolbar)

from matplotlib.figure import Figure
import random
import numpy as np

定义一个matplotlib显示的类MplWidget

class MplWidget(QWidget):

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

        self.canvas = FigureCanvas(Figure())

        vertical_layout = QVBoxLayout()
        vertical_layout.addWidget(self.canvas)
        vertical_layout.addWidget(NavigationToolbar(self.canvas, self))

        self.canvas.axes = self.canvas.figure.add_subplot(111)
        self.setLayout(vertical_layout)

加载ui文件并导入MplWidget类

class MainWidget(QWidget):


    def __init__(self):
        QWidget.__init__(self)

        designer_file = QFile("qt_designer.ui")
        designer_file.open(QFile.ReadOnly)

        loader = QUiLoader()
        loader.registerCustomWidget(MplWidget)
        self.ui = loader.load(designer_file, self)

        designer_file.close()

        self.ui.button1.clicked.connect(self.update_graph)

        self.setWindowTitle("PySide2 & Matplotlib Example GUI")

        grid_layout = QGridLayout()
        grid_layout.addWidget(self.ui)
        self.setLayout(grid_layout)


    def update_graph(self):
        fs = 500
        f = random.randint(1, 100)
        ts = 1 / fs
        length_of_signal = 100
        t = np.linspace(0, 1, length_of_signal)

        cosinus_signal = np.cos(2 * np.pi * f * t)
        sinus_signal = np.sin(2 * np.pi * f * t)

        self.ui.asss.canvas.axes.clear()
        self.ui.asss.canvas.axes.plot(t, cosinus_signal)
        self.ui.asss.canvas.axes.plot(t, sinus_signal)
        self.ui.asss.canvas.axes.legend(('cosinus', 'sinus'), loc='upper right')
        self.ui.asss.canvas.axes.set_title('Cosinus - Sinus Signals')
        self.ui.asss.canvas.draw()

ui文件编写

打开qt designer

1.新建窗体

在这里插入图片描述

2.添加一个pushbutton一个widget

在这里插入图片描述

按照代码编写的button应命名为(button1),widget应命名为(asss) #括号内内容可以按照需求更改

3.重点,选中widget并右键选择提升为,英文版为promote啥的
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

添加一个提升类命名为MplWidget并选择提升见对象查看器中asss被提升为MplWidget

在这里插入图片描述

4.保存ui

5.ui和.py程序放在同一目录

将该完整代码运行

from PySide2.QtWidgets import *
from PySide2.QtUiTools import QUiLoader
from PySide2.QtCore import QFile
from matplotlib.backends.backend_qt5agg import (
    FigureCanvasQTAgg as FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
from matplotlib.figure import Figure
import random
import numpy as np

class MplWidget(QWidget):

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

        self.canvas = FigureCanvas(Figure())

        vertical_layout = QVBoxLayout()
        vertical_layout.addWidget(self.canvas)
        vertical_layout.addWidget(NavigationToolbar(self.canvas, self))

        self.canvas.axes = self.canvas.figure.add_subplot(111)
        self.setLayout(vertical_layout)
class MainWidget(QWidget):


    def __init__(self):
        QWidget.__init__(self)

        designer_file = QFile("qt_designer.ui")
        designer_file.open(QFile.ReadOnly)

        loader = QUiLoader()
        loader.registerCustomWidget(MplWidget)
        self.ui = loader.load(designer_file, self)

        designer_file.close()

        self.ui.button1.clicked.connect(self.update_graph)

        self.setWindowTitle("PySide2 & Matplotlib Example GUI")

        grid_layout = QGridLayout()
        grid_layout.addWidget(self.ui)
        self.setLayout(grid_layout)


    def update_graph(self):
        fs = 500
        f = random.randint(1, 100)
        ts = 1 / fs
        length_of_signal = 100
        t = np.linspace(0, 1, length_of_signal)

        cosinus_signal = np.cos(2 * np.pi * f * t)
        sinus_signal = np.sin(2 * np.pi * f * t)

        self.ui.asss.canvas.axes.clear()
        self.ui.asss.canvas.axes.plot(t, cosinus_signal)
        self.ui.asss.canvas.axes.plot(t, sinus_signal)
        self.ui.asss.canvas.axes.legend(('cosinus', 'sinus'), loc='upper right')
        self.ui.asss.canvas.axes.set_title('Cosinus - Sinus Signals')
        self.ui.asss.canvas.draw()

app = QApplication([])
window = MainWidget()
window.show()
app.exec_()

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

大功告成!!!

Logo

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。

更多推荐