pyqt5+matplotlib绘制动态双y轴曲线

pyqt5+matplotlib绘制动态双y轴曲线

本篇使用pyqt5+matplotlib绘制双轴曲线,效果图如下:
动态曲线



前言

需要用到pyqt5里面的GroupBox控件。
需要用到matplotlib库中的 Line2D类


一、首先画简单pyqt界面

创建一个groupBox控件

class linewidget(QMainWindow):
    def __init__(self):
        super(linewidget, self).__init__()
        self.setWindowTitle('绘制动态曲线')
        self.resize(1000, 800)

        # 创建一个groupbox, 用来画动态曲线
        self.groupBox = QtWidgets.QGroupBox(self)
        self.groupBox.setGeometry(QtCore.QRect(100, 200, 800, 300))

二、创建画板类

class Figure_Canvas(FigureCanvas):
    """
    创建画板类
    """

    def __init__(self, width=3.2, height=2.7):
        self.fig = Figure(figsize=(width, height), dpi=70)
        super(Figure_Canvas, self).__init__(self.fig)
        self.ax = self.fig.add_subplot(111)  # 111表示1行1列,第一张曲线图

三、绘制曲线

在主界面添加函数

    def load_line(self):
        self.LineFigure = Figure_Canvas()
        self.LineFigureLayout = QGridLayout(self.groupBox)
        self.LineFigureLayout.addWidget(self.LineFigure)

        # 准备数据,绘制曲线
        x_data = np.arange(-4, 4, 0.02)
        y_data = np.sin(x_data)
        y2_data = np.cos(x_data)
        self.LineFigure.add_line(x_data, y_data, y2_data)

在画板类中添加函数

    def add_line(self, x_data, y_data, y2_data=None):
        self.line = Line2D(x_data, y_data)  # 绘制2D折线图

        # ------------------调整折线图基本样式---------------------#

        # self.line.set_ls('--')  # 设置连线
        # self.line.set_marker('*') # 设置每个点
        # self.line.set_color('red')  # 设置线条颜色

        self.ax.grid(True)  # 添加网格
        self.ax.set_title('动态曲线')  # 设置标题

        # 设置xy轴最大最小值,找到x_data, y_data最大最小值
        self.ax.set_xlim(np.min(x_data), np.max(x_data))
        self.ax.set_ylim(np.min(y_data), np.max(y_data) + 2)  # y轴稍微多一点,会好看一点

        self.ax.set_xlabel('x坐标')  # 设置坐标名称
        self.ax.set_ylabel('y坐标')

        # 在曲线下方填充颜色
        # self.ax.fill_between(x_data, y_data, color='g', alpha=0.1)

        # self.ax.legend([self.line], ['sinx'])  # 添加图例

        # ------------------------------------------------------#
        self.ax.add_line(self.line)


        # 绘制第二条曲线
        self.line2 = Line2D(x_data, y2_data)
        self.ax.add_line(self.line2)
        self.line2.set_color('red')  # 设置线条颜色
        self.ax.legend([self.line, self.line2], ['sinx', 'cosx'])  # 添加图例

        self.ax2 = self.ax.twinx()
        self.ax2.set_ylabel('y2坐标')

四、创建定时器,动态更新数据

# 创建定时器,使曲线图动态更新
        self.timer = QTimer()
        self.timer.start(10)
        self.ts = time.time()
        self.timer.timeout.connect(self.Updatedata)
    def Updatedata(self):
        dt = time.time() - self.ts
        x_data = np.arange(-4, 4, 0.02)
        z_data = np.sin(x_data + dt)  # 准备动态数据

        h_data = np.cos(x_data + dt)

        self.LineFigure.line.set_ydata(z_data)  # 更新数据
        self.LineFigure.line2.set_ydata(h_data)

        self.LineFigure.draw()  # 重新画图

五、完整代码如下

import sys

import time
from PyQt5 import QtWidgets, QtCore
from PyQt5.QtCore import QTimer

from PyQt5.QtWidgets import QMainWindow, QApplication, QGridLayout

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
from matplotlib.lines import Line2D
import matplotlib

import numpy as np

matplotlib.rcParams['font.sans-serif'] = ['SimHei']  # 解决坐标轴中文显示问题
matplotlib.rcParams['axes.unicode_minus'] = False  # 解决负号不显示的问题


class Figure_Canvas(FigureCanvas):
    """
    创建画板类
    """

    def __init__(self, width=3.2, height=2.7):
        self.fig = Figure(figsize=(width, height), dpi=70)
        super(Figure_Canvas, self).__init__(self.fig)
        self.ax = self.fig.add_subplot(111)  # 111表示1行1列,第一张曲线图

    def add_line(self, x_data, y_data, y2_data=None):
        self.line = Line2D(x_data, y_data)  # 绘制2D折线图

        # ------------------调整折线图基本样式---------------------#

        # self.line.set_ls('--')  # 设置连线
        # self.line.set_marker('*') # 设置每个点
        # self.line.set_color('red')  # 设置线条颜色

        self.ax.grid(True)  # 添加网格
        self.ax.set_title('动态曲线')  # 设置标题

        # 设置xy轴最大最小值,找到x_data, y_data最大最小值
        self.ax.set_xlim(np.min(x_data), np.max(x_data))
        self.ax.set_ylim(np.min(y_data), np.max(y_data) + 2)  # y轴稍微多一点,会好看一点

        self.ax.set_xlabel('x坐标')  # 设置坐标名称
        self.ax.set_ylabel('y坐标')

        # 在曲线下方填充颜色
        # self.ax.fill_between(x_data, y_data, color='g', alpha=0.1)

        # self.ax.legend([self.line], ['sinx'])  # 添加图例

        # ------------------------------------------------------#
        self.ax.add_line(self.line)


        # 绘制第二条曲线
        self.line2 = Line2D(x_data, y2_data)
        self.ax.add_line(self.line2)
        self.line2.set_color('red')  # 设置线条颜色
        self.ax.legend([self.line, self.line2], ['sinx', 'cosx'])  # 添加图例

        self.ax2 = self.ax.twinx()
        self.ax2.set_ylabel('y2坐标')


class linewidget(QMainWindow):
    def __init__(self):
        super(linewidget, self).__init__()
        self.setWindowTitle('绘制动态曲线')
        self.resize(1000, 800)

        # 创建一个groupbox, 用来画动态曲线
        self.groupBox = QtWidgets.QGroupBox(self)
        self.groupBox.setGeometry(QtCore.QRect(100, 200, 800, 300))

        self.load_line()  # 加载动态曲线

        # 创建定时器,使曲线图动态更新
        self.timer = QTimer()
        self.timer.start(10)
        self.ts = time.time()
        self.timer.timeout.connect(self.Updatedata)

    def load_line(self):
        self.LineFigure = Figure_Canvas()
        self.LineFigureLayout = QGridLayout(self.groupBox)
        self.LineFigureLayout.addWidget(self.LineFigure)

        # 准备数据,绘制曲线
        x_data = np.arange(-4, 4, 0.02)
        y_data = np.sin(x_data)
        y2_data = np.cos(x_data)
        self.LineFigure.add_line(x_data, y_data, y2_data)

    def Updatedata(self):
        dt = time.time() - self.ts
        x_data = np.arange(-4, 4, 0.02)
        z_data = np.sin(x_data + dt)  # 准备动态数据

        h_data = np.cos(x_data + dt)

        self.LineFigure.line.set_ydata(z_data)  # 更新数据
        self.LineFigure.line2.set_ydata(h_data)

        self.LineFigure.draw()  # 重新画图


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

总结

提示:

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

方寸之间 

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值