PyQtGraph – 散点图上不同颜色的点

PyQtGraph是一个用于科学计算的Python库，其中包括了绘图工具。本文将介绍如何使用PyQtGraph在散点图上展示不同颜色的点。

安装

在使用之前，我们需要先安装PyQtGraph库。可以使用以下命令进行安装：

pip install pyqtgraph

示例

下面的示例代码展示了如何在散点图上展示不同颜色的点：

import pyqtgraph as pg
import numpy as np

# 生成随机数据
numPoints = 1000
data = np.random.normal(size=(numPoints, 2))

# 根据数据中的y值选择要展示的颜色
colors = np.zeros((numPoints, 3))
colors[data[:, 1] > 0, 0] = 1.0
colors[data[:, 1] <= 0, 2] = 1.0

# 创建绘制窗口
win = pg.plot()
win.setWindowTitle('Scatter plot with colors')

# 绘制散点图
scatter = pg.ScatterPlotItem(pos=data,
                              size=10,
                              pen=pg.mkPen('w'),
                              brush=pg.mkBrush(colors))
win.addItem(scatter)

# 显示窗口
pg.QtGui.QApplication.exec_()

我们首先生成了1000个随机数据点。这些数据点包括x和y两个坐标值。接着，我们根据数据中的y值选择要展示的颜色。如果y值大于0，我们将点的颜色设置为红色（RGB(1, 0, 0)），否则设置为蓝色（RGB(0, 0, 1)）。

接着创建绘图窗口，并在其中绘制了散点图。注意，我们使用了ScatterPlotItem类，并通过brush参数指定了点的颜色。

最后，我们显示了绘图窗口，并等待用户关闭窗口。

结论

本文介绍了如何使用PyQtGraph在散点图中展示不同颜色的点。我们利用了ScatterPlotItem类，并通过颜色数组指定了每个点的颜色。PyQtGraph还提供了其他的绘图工具，可以实现更多不同的可视化效果。