PyQtGraph - 散点图点的旋转数据

PyQtGraph 是一个用于数据可视化的 Python 库，可以绘制各种类型的图形，其中包括散点图。在散点图中，每个点都表示一个数据点，其 x，y 坐标可以表示两个特征的值。在某些情况下，您可能需要通过旋转点来表示第三个特征的值。在本文中，我们将介绍如何在 PyQtGraph 中旋转散点图点并绑定数据。

安装 PyQtGraph

在使用 PyQtGraph 之前，您需要先安装 PyQtGraph 库。您可以通过以下命令在命令行中安装 PyQtGraph：

pip install pyqtgraph

创建散点图

在创建散点图之前，您需要导入必要的库。

import sys
from pyqtgraph.Qt import QtGui, QtCore
import numpy as np
import pyqtgraph as pg

接下来，我们将使用 numpy 生成一些随机数据，并将它们传递给散点图：

# Generate some random data
x = np.random.normal(size=100)
y = np.random.normal(size=100)

# Create the scatter plot
win = pg.GraphicsWindow(title='Scatter plot')
plot = win.addPlot()
scatter = pg.ScatterPlotItem(x=x, y=y, pen=None, brush='r', size=10)
plot.addItem(scatter)

这将创建一个散点图窗口，并将散点图添加到其中。现在，我们将为每个点设置角度，该角度将使用第三个特征的值进行计算。

散点图点的旋转数据

要旋转散点图点，您需要为每个点定义一个角度。在本例中，我们将使用 numpy 的 arctan2() 函数计算每个点的角度，并将其设置为散点图的 angle 属性。我们还将为每个点添加一个额外的属性 data，以便将第三个特征的数据绑定到每个点。

# Add rotation angle and extra data for each point
angles = np.arctan2(y, x) * 180 / np.pi
data = np.random.normal(size=100)
scatter.setData(pos=np.column_stack([x, y]), size=10, pen=None, brush='r', 
                symbol='o', symbolBrush='r', symbolSize=10, 
                data=np.column_stack([data, angles, np.ones((100, 2))]))
scatter.setOpts(antialias=True, pxMode=False, name='Scatter')

在上面的代码中，我们使用 column_stack() 函数将 x，y，角度和额外的数据垂直组合，以便我们可以将它们传递给散点图。 setData() 方法将数据设置为散点图，并设置其颜色，大小和符号。我们还通过将 pxMode 参数设置为 False，使用矢量图绘制散点图。

这将为散点图中的每个点设置角度，并将其绑定到额外的数据。现在，我们可以运行程序并看到旋转的散点图。

完整代码

import sys
from pyqtgraph.Qt import QtGui, QtCore
import numpy as np
import pyqtgraph as pg

# Generate some random data
x = np.random.normal(size=100)
y = np.random.normal(size=100)

# Create the scatter plot
win = pg.GraphicsWindow(title='Scatter plot')
plot = win.addPlot()
scatter = pg.ScatterPlotItem(x=x, y=y, pen=None, brush='r', size=10)
plot.addItem(scatter)

# Add rotation angle and extra data for each point
angles = np.arctan2(y, x) * 180 / np.pi
data = np.random.normal(size=100)
scatter.setData(pos=np.column_stack([x, y]), size=10, pen=None, brush='r', 
                symbol='o', symbolBrush='r', symbolSize=10, 
                data=np.column_stack([data, angles, np.ones((100, 2))]))
scatter.setOpts(antialias=True, pxMode=False, name='Scatter')

# Run the program
if __name__ == '__main__':
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        QtGui.QApplication.instance().exec_()

总结

在 PyQtGraph 中旋转散点图点，可以显示第三个特征的值，同时不影响已有数据的可视化。我们可以使用 numpy 的 arctan2() 函数计算每个点的角度，并将其设置为散点图的 angle 属性。我们还将为每个点添加一个额外的属性 data，以便将第三个特征的数据绑定到每个点。