PyQtGraph - 设置散点图的点

PyQtGraph是一个用于数据可视化的Python库，它可以绘制2D图表、3D图表、散点图、直方图、波形等多种图表类型。本篇文章将介绍如何使用PyQtGraph库设置散点图的点。

安装PyQtGraph

可以使用pip来安装PyQtGraph库，命令如下：

pip install PyQtGraph

绘制散点图

首先我们需要导入PyQtGraph库，然后创建QApplication实例，并创建一个窗口用于显示散点图。

import sys
import numpy as np
import pyqtgraph as pg
from PyQt5.QtWidgets import QApplication, QMainWindow

app = QApplication(sys.argv)
mainwindow = QMainWindow()
mainwindow.setWindowTitle('Scatter Plot')
mainwindow.setGeometry(100, 100, 800, 600)
mainwindow.show()

graph_widget = pg.GraphicsLayoutWidget(parent=mainwindow)
mainwindow.setCentralWidget(graph_widget)

plot_widget = graph_widget.addPlot()

然后我们可以生成一些虚拟数据用于填充散点图。

x = np.linspace(-10, 10, 100)
y = np.sin(x)
size = np.random.rand(100) * 20
color = np.random.rand(100)

接下来我们可以用生成的数据绘制散点图。

scatter_item = pg.ScatterPlotItem(x=x, y=y, size=size, brush=color)
plot_widget.addItem(scatter_item)

这样我们就可以在窗口上看到一个有多个点的散点图了。

设置散点图的点

我们可以通过修改x、y、size、color等变量来改变散点图的点。

例如我们可以将所有点的颜色都设为红色。

color = (1, 0, 0)

或者我们可以将所有点的大小都设为10。

size = 10

我们也可以针对每个点设置不同的颜色和大小。

color = np.random.rand(100)
size = np.random.rand(100) * 20

如果我们想要将某些点突出显示，我们只需要将其单独设定颜色和大小即可。

color[0] = (1, 0, 0)
size[0] = 30

除了可以设置点的大小和颜色以外，我们还可以设置点的形状。

symbol = ['o', 't', 's', 'd', 'p', '+', 'x'][index]
scatter_item = pg.ScatterPlotItem(x=x, y=y, size=size, symbol=symbol, brush=color)

其中symbol参数可以设置点的形状，可以设置为圆形（'o'）、正方形（'s'）、菱形（'d'）等等。

以上就是PyQtGraph设置散点图的点的介绍，希望对大家有所帮助。完整代码如下：