python pyside2的matplotlib模块导入

将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_()

大功告成！！！