如何在 PyQtGraph 的一个图中绘制两个实时数据？

Posted 2023-03-25

【中文标题】如何在 PyQtGraph 的一个图中绘制两个实时数据？

【英文标题】：How to plot two real-time data in one single plot in PyQtGraph?

【发布时间】：2017-03-27 09:40:38

【问题描述】：

我愿意获取 2 个随机数据并使用 PyQtGraph 以实时方式将其绘制在同一个 Widget 中。我希望它们显示为红点和蓝点。但是，经过一段时间的努力，我的脚本无法正常工作。

我想知道如何才能在同一个图中获取两个数据。

我知道这是一个愚蠢的问题。我是 Python 和编码的初学者。

这是我的代码：

#-*- coding: utf-8 -*-
import random
import time
from collections import deque
import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui
import numpy as np
import os
import spidev

win = pg.GraphicsWindow()
win.setWindowTitle('DOTS')


p1 = win.addPlot()
p1.setRange(yRange=[0,25])
p1.setRange(xRange=[0,25])
curve1 = p1.plot()


nsamples=300 #Number of lines for the data

dataRed= np.zeros((nsamples,2),float) #Matrix for the Red dots
dataBlue=np.zeros((nsamples,2),float) #Matrix for the Blue dots

def getData():
    global dataRed, dataBlue

    t0= random.uniform(1.6,20.5) #Acquiring Data
    d0= random.uniform(1.6,20.5) #Acquiring Data
    vec=(t0, d0)

    dataRed[:-1] = dataRed[1:]
    dataRed[-1]=np.array(vec)

    t0= random.uniform(1.6,20.5) #Acquiring Data
    d0= random.uniform(1.6,20.5) #Acquiring Data
    vec=(t0, d0)

    dataBlue[:-1] = dataBlue[1:]
    dataBlue[-1]=np.array(vec)


def plot():

    #Blue Dots
    curve1.setData(dataBlue, pen=None, symbol='o', symbolPen=None, symbolSize=4, symbolBrush=('b'))
    #Red Dots
    curve1.setData(dataRed, pen=None, symbol='o', symbolPen=None, symbolSize=4, symbolBrush=('r'))   


def update():

    getData()
    plot()

timer = pg.QtCore.QTimer()
timer.timeout.connect(update)
timer.start(50)

## Start Qt event loop unless running in interactive mode or using pyside.
if __name__ == '__main__':
    import sys
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        QtGui.QApplication.instance().exec_()# -*- coding: utf-8 -*-

非常感谢您的帮助。

【问题讨论】：

dataRed[:-1] = dataRed[1:] dataRed[-1]=np.array(vec) 这不是将数据 1 位移到后面，然后替换最后一项而不是第一项吗？

@MaartenFabré 我本来打算这样做的，实际上这是一个与实时曲线相关的更大问题的简单版本。不管怎样，谢谢你的评论！

【参考方案1】：

虽然从功能的角度来看 pyqtgraph 是一个很棒的包，但遗憾的是缺少文档，你真的只需要深入研究代码并开始了解对象的结构。

当你打电话时：

p1 = win.addPlot() 

这将返回对 PlotItem 的引用，此时您现在可以将多个 PlotDataItems 添加到此 p1 对象（请参阅此处的半有用结构图：http://www.pyqtgraph.org/documentation/plotting.html#organization-of-plotting-classes）

所以当你打电话时：

curve1 = p1.plot()

这会添加 PlotDataItem #1，...您现在需要再次调用它以获取第二个要使用的引用：

curve2 = p1.plot()

这成为 PlotDataItem #2，然后您可以将其用于 plot() 方法中的第二个 setData 方法，以便在 update() 期间调用。看起来像：

def plot():

    #Blue Dots
    curve1.setData(dataBlue, pen=None, symbol='o', symbolPen=None, symbolSize=4, symbolBrush=('b'))
    #Red Dots
    curve2.setData(dataRed, pen=None, symbol='o', symbolPen=None, symbolSize=4, symbolBrush=('r')) 

【讨论】：

哇！非常感谢。我的问题解决了。非常感谢您的所有解释！