📜  PyQtGraph - 获取散点图的位置(1)

📅  最后修改于: 2023-12-03 15:18:50.822000             🧑  作者: Mango

PyQtGraph - 获取散点图的位置

PyQtGraph是一个Python图形库,可以用于生成科学图表、特别是实时数据可视化。

在PyQtGraph中获取散点图的位置是很简单的。

import pyqtgraph as pg

app = QtGui.QApplication([])
plt = pg.plot()
x = [1, 2, 3, 4, 5, 6]
y = [1, 4, 2, 3, 5, 6]
scatter = pg.ScatterPlotItem(x=x, y=y)
plt.addItem(scatter)

# 获取一个散点的位置
print(scatter.pos())

# 获取所有散点的位置
positions = scatter.data
print(positions)

在上述代码中,我们使用ScatterPlotItem为绘图区域添加了一个散点图。使用scatter.pos()可以获取散点的位置,使用scatter.data可以获取所有散点的位置。

返回的数据是numpy数组,可以使用numpy模块的函数进一步处理。

import numpy as np

# 获取所有散点的横坐标
x_pos = np.array(scatter.data['pos'])[:, 0]
print(x_pos)

# 获取所有散点的纵坐标
y_pos = np.array(scatter.data['pos'])[:, 1]
print(y_pos)

在上述代码中,我们使用numpy的array函数将scatter.data['pos']转化为numpy数组。使用[:, 0]获取所有散点的横坐标,使用[:, 1]获取所有散点的纵坐标。