PyQtGraph - 在散点图中获取点的像素填充

PyQtGraph是一个Python库，提供了丰富的图形绘制功能，包括线图、散点图等。本文将介绍如何在散点图中获取点的像素填充。

安装PyQtGraph

在开始介绍之前，需要安装PyQtGraph库。可以通过pip工具进行安装：

pip install pyqtgraph

创建散点图

在创建散点图之前，需要准备一些数据。本例中，将生成100个随机点，并用不同的颜色将它们填充:

import pyqtgraph as pg
import numpy as np

# 创建随机数据
np.random.seed(12345)
x = np.random.normal(size=100)
y = np.random.normal(size=100)
colors = np.random.randint(0, 256, size=(100, 3))

# 创建图形窗口
win = pg.GraphicsWindow(title="散点图")
plot = win.addPlot(title="散点图")

# 绘制散点图
scatter = pg.ScatterPlotItem(x=x, y=y, brush=colors, size=10)
plot.addItem(scatter)

获取像素填充

要获取散点图中单个点的像素填充，需要使用item的方法。具体来说，需要定义一个回调函数，它将在单击散点图上的点时被调用。该函数将获取当前数据点的索引，并使用item的方法获取与该索引关联的像素颜色：

def on_plot_click(evt):
    # 获取鼠标点击位置相对于散点图的坐标
    pos = plot.vb.mapSceneToView(evt[0])
    
    # 计算最近的数据点
    dists = np.sqrt((x - pos.x())**2 + (y - pos.y())**2)
    idx = np.argmin(dists)
    
    # 获取该数据点的像素填充颜色
    brush = scatter.points()[idx].brush()
    
    print(f"单击了点({x[idx]}, {y[idx]}), 颜色为{brush.color().getRgb()}。")
    
plot.scene().sigMouseClicked.connect(on_plot_click)

在上面的代码中，首先获取鼠标点击位置相对于散点图的坐标。然后计算最近的数据点，并获取其索引。最后，通过查询item的方法获取该数据点的像素填充颜色，并打印出来。

完整代码

以下是完整的示例代码：

import pyqtgraph as pg
import numpy as np


def on_plot_click(evt):
    # 获取鼠标点击位置相对于散点图的坐标
    pos = plot.vb.mapSceneToView(evt[0])
    
    # 计算最近的数据点
    dists = np.sqrt((x - pos.x())**2 + (y - pos.y())**2)
    idx = np.argmin(dists)
    
    # 获取该数据点的像素填充颜色
    brush = scatter.points()[idx].brush()
    
    print(f"单击了点({x[idx]}, {y[idx]}), 颜色为{brush.color().getRgb()}。")


# 创建随机数据
np.random.seed(12345)
x = np.random.normal(size=100)
y = np.random.normal(size=100)
colors = np.random.randint(0, 256, size=(100, 3))

# 创建图形窗口
win = pg.GraphicsWindow(title="散点图")
plot = win.addPlot(title="散点图")

# 绘制散点图
scatter = pg.ScatterPlotItem(x=x, y=y, brush=colors, size=10)
plot.addItem(scatter)

# 为plot添加单击事件
plot.scene().sigMouseClicked.connect(on_plot_click)

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

总结

通过PyQtGraph，可以轻松地创建散点图，并获取散点图中单个点的像素填充颜色。此外，PyQtGraph还提供了很多其他的图形绘制功能，可供开发者使用。