我试图使用pyqtgraph生成散点图和直方图的矩阵。每个散点图的输入（x和y值）都是长度大于1,000,000的numpy数组。生成这些图需要很长时间（2x2矩阵需要超过1分钟），而matplotlib在生成相同类型的图时速度更快。有没有什么方法可以加快速度？下面是我使用的代码。
谢谢。
谢谢。
from pyqtgraph.Qt import QtGui, QtCore
import numpy as np
import pyqtgraph as pg
def scatter_matrix(data, cols):
pg.setConfigOption('background', 'w')
pg.setConfigOption('foreground', 'k')
now = pg.ptime.time()
app = QtGui.QApplication([])
win = pg.GraphicsWindow(title="Scater Plot Matrix")
win.resize(800,600)
for i, col_i in enumerate(cols):
for j, col_j in enumerate(cols):
x = data[col_i]
y = data[col_j]
if i == j:
current_plot = win.addPlot(title="Histogram")
y,x = np.histogram(x, bins=100)
curve = pg.PlotCurveItem(x, y, stepMode=True, fillLevel=0, brush=(0, 0, 255, 80))
current_plot.addItem(curve)
else:
current_plot = win.addPlot(title="Scatter plot")
current_plot.plot(x, y, pen=None, symbol='t', symbolPen=None, symbolSize=10, symbolBrush=(100, 100, 255, 50))
current_plot.setLabel('left', "{}".format(col_i), units='')
current_plot.setLabel('bottom', "{}".format(col_j), units='')
current_plot.setLogMode(x=False, y=False)
win.nextRow()
## Start Qt event loop unless running in interactive mode or using pyside.
import sys
print "Plot time: %0.2f sec" % (pg.ptime.time()-now)
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
app.exec_()
data = {}
for key in ['a','b']:
data[key] = np.random.normal(size=(1000000), scale=1e-5)
scatter_matrix(data,['a','b'])
在网上搜索了很多后，我最终尝试了基于GPU的绘图库Galry。结果加速超过了100倍！以下是代码。虽然如此，我仍然想知道是否有使用PyQtGraph加速绘图的方法。
import numpy as np
from galry import *
import time
class MyPaintManager(PlotPaintManager):
def initialize(self):
if self.parent.visual == BarVisual:
self.add_visual(self.parent.visual, self.parent.x, primitive_type= self.parent.plot_type, color='b')
elif self.parent.visual == PlotVisual:
self.add_visual(self.parent.visual, x=self.parent.x, y=self.parent.y, primitive_type= self.parent.plot_type, color='b')
class MyWidget(GalryWidget):
def initialize(self, x, y, visual, title=None, plot_type=None):
self.activate_grid = True
self.show_grid = True
self.x = x
self.y = y
self.visual = visual
self.plot_type = plot_type
self.title = title
self.set_bindings(PlotBindings)
self.set_companion_classes(
paint_manager=MyPaintManager,
interaction_manager=PlotInteractionManager,)
self.initialize_companion_classes()
def scatter_matrix(df, cols):
now = time.time()
class Window(QtGui.QWidget):
def __init__(self):
super(Window, self).__init__()
self.initUI()
def initUI(self):
vbox = QtGui.QGridLayout()
self.setLayout(vbox)
self.setGeometry(300, 300, 600, 600)
for i, col_i in enumerate(cols):
for j, col_j in enumerate(cols):
x = df[col_i]
y = df[col_j]
if i == j:
y,x = np.histogram(x, bins=100)
vbox.addWidget(MyWidget(x=y,y=y, visual = BarVisual, title='{}_vs_{}'.format(col_i, col_j)), i, j)
else:
vbox.addWidget(MyWidget(x=x,y=y, visual = PlotVisual, title='{}_vs_{}'.format(col_i, col_j), plot_type='POINTS'), i, j)
print "Plot time: %0.2f sec" % (time.time()-now)
self.show()
show_window(Window)
if __name__ == '__main__':
data = {}
for key in ['a','b']:
data[key] = np.random.normal(size=(1000000), scale=1e-5)
scatter_matrix(data,['a','b'])