总结一下PyQT5 + matplotlib的使用方法

时间：2024-05-22 09:36:48

搜了一下网上的用法还挺多的，买的这本PyQt5快速开发和实战里也有介绍，尝试了一下还是可以做到的。

原理是使用：

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas

获取matplotlib下backend-qt5agg 接口

（其实我也不懂这其中的细节，反正能实现自己想要的功能了）

 

首先是封装一个FigureCanvasQTAgg类，画图的时候调用这个类就行了

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
import matplotlib.pyplot as plt
from PyQt5.QtWidgets import QSizePolicy


class Mydemo(FigureCanvas):

    def __init__(self, parent=None, width=5, height=4, dpi=100):

        plt.rcParams['font.family'] = ['SimHei']    # 更换字体使中文显示正常
        plt.rcParams['axes.unicode_minus'] = False    # 用来正常显示负号
        
        # 创建绘图对象，就像matlab里创建一个新的窗口那样

        self.fig = Figure(figsize=(width, height), dpi=dpi) 

   
        # 紧接着在这个新的窗口加上一个子图，也就是实际画图的地方
        # 看到subplot我想你也明白，可以在这里添加多个子图，并且规定子图的位置，       
        # 方法和matlab的subplot一摸一样，为了演示我加了一个新的axes1

        self.axes = self.fig.add_subplot(2, 2, 1)
        self.axes1 = self.fig.add_subplot(2, 2, 4)

        
        # 下面这个是画新图后不保留上次的图形，但这个代码似乎有问题报错了，先注释掉
        # self.axes.hold(False) 

        FigureCanvas.__init__(self, self.fig)
        self.setParent(parent)


        '''定义FigureCanvas的尺寸策略，这部分的意思是设置FigureCanvas，使之尽可能的向外填充空间。'''
        FigureCanvas.setSizePolicy(self, 
                                        QSizePolicy.Expanding, 
                                        QSizePolicy.Expanding)
        FigureCanvas.updateGeometry(self)

 

在PyQt5中需要用到布局管理器，也就是

from PyQt5.QtWidgets import QHBoxLayout, QGridLayout, QVBoxLayout, QFormLayout  这4个中的任意一个。

通过Layout.addWidget来将上面的FigureCanvasQTAgg类对象添加进去

下图提供一个演示的例子

# 最重要的一件事，在所有的模块导入之前第一件要做的事就是使用Qt5Agg接口
import matplotlib
matplotlib.use("Qt5Agg")   

from PyQt5.QtWidgets import QApplication, QVBoxLayout, QWidget



class Try(QWidget):
    
    def __init__(self):
        super().__init__()
        self.initUi()
    
    def initUi(self):
        self.layout = QVBoxLayout(self)    # 创建一个垂直布局layout
         
        
        # 实例化一个FigureCanvasQTAgg对象self.figure1

        self.figure1 = Mydemo(width=5, height=4, dpi=100) 
        

        # 将实例化对象self.figure1添加到垂直布局layout中

        self.layout.addWidget(self.figure1)
        

        # 在对象的axes和axes1画板上作图
        data = list(range(1000))
        self.figure.axes.plot(data)
        self.figure.axes1.plot(data)

这样就完成任务了，通过测试代码就可以看到效果，下面的代码是完整的，可以直接复制运行

import sys
import matplotlib
matplotlib.use("Qt5Agg")
from PyQt5.QtWidgets import QApplication, QVBoxLayout, QSizePolicy, QWidget
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
import matplotlib.pyplot as plt


class Mydemo(FigureCanvas):

    def __init__(self, parent=None, width=5, height=4, dpi=100):

        plt.rcParams['font.family'] = ['SimHei']
        plt.rcParams['axes.unicode_minus'] = False

        self.fig = Figure(figsize=(width, height), dpi=dpi)
        self.axes = self.fig.add_subplot(2, 1, 1)
        self.axes1 = self.fig.add_subplot(2, 1, 2)

        FigureCanvas.__init__(self, self.fig)
        self.setParent(parent)

        FigureCanvas.setSizePolicy(self,
                                   QSizePolicy.Expanding,
                                   QSizePolicy.Expanding)
        FigureCanvas.updateGeometry(self)


class Try(QWidget):
    def __init__(self):
        super().__init__()
        self.initUi()

    def initUi(self):
        self.layout = QVBoxLayout(self)
        self.figure1 = Mydemo(width=5, height=4, dpi=100)
        self.layout.addWidget(self.figure1)
        data = list(range(1000))
        self.figure1.axes.plot(data)
        self.figure1.axes1.plot(data)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    ui = Try()
    ui.show()
    sys.exit(app.exec_())

最终结果是这样的：

总结一下PyQT5 + matplotlib的使用方法

 

非常有意思，一个figure可以有很多歌axes，只要你位置规划好了。你也可以在Qtdesigner里先设计好UI界面，然后通过常规操作进行调用。只需要给matplotlib准备一个布局就行了。

相关文章