PyQtGraph - 设置散点图每个点的大小

PyQtGraph 是一个用于数据可视化的强大工具，支持绘制各种类型的图表，包括散点图。散点图广泛用于显示两个变量之间的关系。在 PyQtGraph 中，我们可以通过设置每个点的大小来进一步突出显示数据的特征。

下面是如何在 PyQtGraph 中设置散点图的每个点的大小的示例代码。

安装 PyQtGraph

要使用 PyQtGraph，首先需要确保已经安装了它。可以使用 pip 包管理器进行安装：

pip install pyqtgraph

创建散点图

在代码中导入pyqtgraph模块，并创建一个绘图窗口和一个绘图区域：

import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui

# 创建绘图窗口和绘图区域
app = QtGui.QApplication([])
win = pg.GraphicsWindow(title="散点图")
plot = win.addPlot()

生成散点图的数据

为了演示设置每个点的大小，我们需要生成一些散点图的随机数据。具体的数据生成方法可以根据实际需求自行调整。下面是一个简单的示例，生成了100个随机点的 x、y 坐标和大小：

import numpy as np

# 生成随机数据
n = 100
x = np.random.normal(size=n)  # x 坐标
y = np.random.normal(size=n)  # y 坐标
size = np.random.normal(size=n)  # 每个点的大小

绘制散点图

现在我们有了数据，接下来可以使用scatter函数在绘图区域中绘制散点图：

# 绘制散点图
scatter = pg.ScatterPlotItem(size=size, pen=pg.mkPen(None), brush=pg.mkBrush(255, 255, 255, 120))
plot.addItem(scatter)
scatter.setData(x=x, y=y)

在上面的代码中，我们使用了ScatterPlotItem类创建一个散点图。size参数用于设置每个点的大小，其中size列表中的每个元素对应一个点的大小。pen参数控制点的边框颜色，mkPen(None)表示无边框。brush参数控制点的填充颜色，mkBrush(255, 255, 255, 120)表示白色半透明。

最后，通过setData方法设置散点图的 x 和 y 坐标。

运行程序

最后，我们需要运行程序，显示绘图窗口和绘制出的散点图：

# 运行程序
QtCore.QCoreApplication.instance().exec_()

结论

通过上述步骤，您可以在 PyQtGraph 中设置散点图每个点的大小。根据需要调整随机数据的生成方法，以及选择合适的点的边框颜色和填充颜色，可以实现更丰富的可视化效果。

注意：上面的代码仅为示例，您可以根据实际情况进行修改和调整。

希望这个介绍对您有所帮助！