Python 如何使用PyQtGraph和ScatterPlotItem（>；1 fps）快速更新图形？

各位PyQtGraph用户好

我试图在pg.ScatterPlotItem中使用Matplotlibs jet颜色，以便使用数据记录器中的数据进行实时图形更新。我在以下几行中将jet颜色转换为RGBA代码：

z=np.random.randint(0,10000,10000) #random z value
norm = mpl.colors.Normalize(vmin=min(z), vmax=max(z))
m = cm.ScalarMappable(norm=norm, cmap=cm.jet)
colors =  m.to_rgba(z, bytes=True)

我注意到，如果点数为10k或更高，则散点图更新的速度可能是一个问题。例如，在我的笔记本电脑上，如果我运行ScatterplotTestSpeed.py中的示例代码，我的10k点的更新速度为0.16 fps。但是，如果我将列表附加为元组，我可以将更新速度提高三倍，达到0.45 fps：

colors_=[]
for i in colors:
  colors_.append(pg.mkBrush(tuple(i)))

这是我通过简单的试看法发现的一个小技巧，但我想知道是否还有更多这样的技巧来加速fps数

下面是我测试更新速度的全部代码。它主要基于pyqtgraphs示例库中的ScatterPlotSpeedTest.py。感谢您的帮助：-）

---

我想我已经找到了解决问题的办法。ScatterPlotItem似乎太慢，无法使用“笔刷”指定的不同颜色连续重新绘制大量点（>500）。相反，我只是使用plotItem和“symbolBrush”为传入的数据点添加Jet颜色（这将为我提供最初搜索的2.5D绘图）。无论添加多少数据点，绘图更新的速度几乎是瞬间的，100个点的重新绘制速度与10000个点的速度一样快（从用户的角度来看）

为什么从记录器打印数据时需要10000种随机颜色？如果你在不使用这种颜色的情况下进行绘图，你将获得超过帧速率10倍的效果。因为我不能运行你的代码，因为它导入了我没有的东西，所以我写了另一段代码，它以11 fps的速度生成10000个点的散点图，颜色一致；1.0 fps，10000种随机颜色。顺便说一句<就速度而言，代码>散点图可能不是最佳选择。例如，在我前几天回答的问题中，我们看到，如果

ImageItem

为100 x 100=10000像素，则可以达到250 fps。这不是10k随机颜色，而是由于z值而产生的喷射着色。每次我从数据记录器中获得新的z值时，我都需要为所有已记录的现有点重新着色。一种解决方案是将jet颜色设置为固定的z值，但颜色将不会完全动态。为了运行代码，您需要从PyQtGraph文件夹中的示例库中运行它，然后将正确加载所有必需的ScatterPlotSpeedTest模板（我想是其中的3个）。总之，到目前为止，这只是一个纯理论的例子，因为实际上没有人能够从10万个样本的散点图中获得很多信息。因此，为了提高速度，最好朝着用例的方向思考，因为（如您所说）值不再是随机的。对于非随机值，可能会有更多的技巧。

#!/usr/bin/python
# -*- coding: utf-8 -*-
"""
For testing rapid updates of ScatterPlotItem under various conditions.

(Scatter plots are still rather slow to draw; expect about 20fps)
"""



## Add path to library (just for examples; you do not need this)
import initExample

import matplotlib as mpl
import matplotlib.cm as cm
from pyqtgraph.Qt import QtGui, QtCore, USE_PYSIDE
import numpy as np
import pyqtgraph as pg
from pyqtgraph.ptime import time
#QtGui.QApplication.setGraphicsSystem('raster')
app = QtGui.QApplication([])
#mw = QtGui.QMainWindow()
#mw.resize(800,800)
if USE_PYSIDE:
    from ScatterPlotSpeedTestTemplate_pyside import Ui_Form
else:
    from ScatterPlotSpeedTestTemplate_pyqt import Ui_Form

win = QtGui.QWidget()
win.setWindowTitle('pyqtgraph example: ScatterPlotSpeedTest')
ui = Ui_Form()
ui.setupUi(win)
win.show()

p = ui.plot
p.setRange(xRange=[-500, 500], yRange=[-500, 500])

data = np.random.normal(size=(50,10000), scale=100)
sizeArray = (np.random.random(500) * 20.).astype(int)
ptr = 0
lastTime = time()
fps = None
def update():
    global curve, data, ptr, p, lastTime, fps
    p.clear()
    if ui.randCheck.isChecked():
        size = sizeArray
    else:
        size = ui.sizeSpin.value()


    z=np.random.randint(0,10000,10000)
    norm = mpl.colors.Normalize(vmin=min(z), vmax=max(z))
    m = cm.ScalarMappable(norm=norm, cmap=cm.jet)
    colors =  m.to_rgba(z, bytes=True)

    colors_=[]
    for i in colors:
      colors_.append(pg.mkBrush(tuple(i)))
      #colors.append(pg.intColor(np.random.randint(0,255), 100))

    curve = pg.ScatterPlotItem(x=data[ptr%50], y=data[(ptr+1)%50],
                   pen='w', brush=colors_, size=size,
                   pxMode=ui.pixelModeCheck.isChecked())


    '''
    curve = pg.ScatterPlotItem(pen='w', size=size, pxMode=ui.pixelModeCheck.isChecked())
    spots3=[]

    for i,j,k in zip(data[ptr%50],data[(ptr+1)%50],colors):
      spots3.append({'pos': (i, j), 'brush':pg.mkBrush(tuple(k))})

    curve.addPoints(spots3)
    '''



    p.addItem(curve)
    ptr += 1
    now = time()
    dt = now - lastTime
    lastTime = now
    if fps is None:
        fps = 1.0/dt
    else:
        s = np.clip(dt*3., 0, 1)
        fps = fps * (1-s) + (1.0/dt) * s
    p.setTitle('%0.2f fps' % fps)
    p.repaint()
    #app.processEvents()  ## force complete redraw for every plot
timer = QtCore.QTimer()
timer.timeout.connect(update)
timer.start(0)



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