📅  最后修改于: 2023-12-03 15:18:50.822000             🧑  作者: Mango
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]
获取所有散点的纵坐标。