我有一个二维的numpy数组如下

import numpy as np
agents = np.array([[  1. , -71.8,  41.2],
       [  1. , -71.8,  41.3],
       [  0. , -71.8,  41.4],
       [  0. , -71.7,  41.4],
       [  0. , -71.6,  41.4]])


现在,我想对其进行可视化。在matplotlib中,我可以执行类似的操作,因为我需要根据第一列的值使用不同的颜色。

import matplotlib.pyplot as plt
plt.scatter(agents [:,1],agents [:,2],c=agents [:,0])


但是如何在pyqtgraph中获得相同的结果?
我试过了

app = QtGui.QApplication(sys.argv)
mw = QtGui.QMainWindow()
mw.resize(800, 800)
view = pg.GraphicsLayoutWidget()
mw.setCentralWidget(view)
mw.setWindowTitle('pyqtgraph example: ScatterPlot')
w1 = view.addPlot()
x = agents[:,1]
y = agents[:,2]
z = agents[:,0]
s = pg.ScatterPlotItem(x, y,pen=pg.mkPen(z))
w1.addItem(s)
mw.show()
sys.exit(QtGui.QApplication.exec_())



得到了这个例外

Exception: Not sure how to make a color from "(array([1., 0., 0., 0.]),)"

最佳答案

如果要为每个点设置颜色,则必须传递颜色列表,在这种情况下,您要传递给它一个列表,如果要设置边框的颜色,则必须传递给QPen;如果需要,则传递给QBrush设置填充颜色。

据我了解,您想要最新的:

s = pg.ScatterPlotItem(x, y, brush=[pg.mkBrush(v) for v in z])
# or
# s = pg.ScatterPlotItem(x, y, brush=list(map(pg.mkBrush, z)))


如果要设置边框颜色,请使用:

s = pg.ScatterPlotItem(x, y, pen=[pg.mkPen(v) for v in z])
# or
# s = pg.ScatterPlotItem(x, y, pen=list(map(pg.mkPen, z)))

10-06 10:33
查看更多