PyQtGraph - 设置散点图的比例

在 PyQtGraph 中，通过 ScatterPlotItem 可以创建散点图。本文将介绍如何设置散点图的比例。

设置比例

要设置散点图的比例，需要使用 setData 函数。setData 函数接受两个参数，第一个参数是横坐标数据，第二个参数是纵坐标数据。如果想要设置每个散点的大小，则需要以列表的形式传递第三个参数。

以下代码展示如何创建一个简单的散点图，并设置每个散点的大小：

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

# 创建一个应用程序对象
app = QtGui.QApplication([])

# 创建一个 GraphicsView 窗口
view = pg.GraphicsView()

# 创建一个场景
scene = pg.GraphicsScene()

# 将场景添加到 GraphicsView 窗口中
view.setCentralItem(scene)

# 创建一个散点图
scatter = pg.ScatterPlotItem()

# 设置数据
x = np.random.normal(size=100)
y = np.random.normal(size=100)
size = np.random.uniform(size=100) * 10
scatter.setData(x=x, y=y, size=size)

# 将散点图添加到场景中
scene.addItem(scatter)

# 显示窗口
view.show()

# 运行应用程序
app.exec_()

在上面的代码中，size 列表中的每个元素乘以了 10，以设置每个散点的大小。您可以尝试修改 size 列表来改变散点的大小。

自定义比例

如果想要自定义比例，可以使用 setSymbolSize 函数。setSymbolSize 函数接受一个浮点数作为参数，用于设置散点的大小。

以下代码展示如何使用 setSymbolSize 函数设置散点图的比例：

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

# 创建一个应用程序对象
app = QtGui.QApplication([])

# 创建一个 GraphicsView 窗口
view = pg.GraphicsView()

# 创建一个场景
scene = pg.GraphicsScene()

# 将场景添加到 GraphicsView 窗口中
view.setCentralItem(scene)

# 创建一个散点图
scatter = pg.ScatterPlotItem()

# 设置数据
x = np.random.normal(size=100)
y = np.random.normal(size=100)
scatter.setData(x=x, y=y)

# 设置比例
scatter.setSymbolSize(10)

# 将散点图添加到场景中
scene.addItem(scatter)

# 显示窗口
view.show()

# 运行应用程序
app.exec_()

在上面的代码中，setSymbolSize 函数设置了散点的大小为 10。您可以尝试修改参数来改变散点的大小。

结论

通过 setData 函数和 setSymbolSize 函数，可以很容易地设置 PyQtGraph 中的散点图的比例。