如何在mplot3d(matplotlib + python)中获取拾取对象的属性

Question

我用mplot3d绘制一系列点：

import pylab as p
import mpl_toolkits.mplot3d.axes3d as p3
fig = p.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter([1], [0], [0], c='r', marker='^', picker=5)
ax.scatter([0], [1], [0], c='g', marker='^', picker=5)
ax.scatter([0], [0], [1], c='b', marker='^', picker=5)

然后我添加一个选择器函数：

def onpick(event):
   ind = event.ind
   print ind
fig.canvas.mpl_connect('pick_event', onpick)

最后剧情：

p.show()

有没有办法从我正在点击的标记中获取 3D 坐标？到目前为止，我可以在 ax.scatter() 上使用的列表中获得该点的索引，但这不会削减它，因为我多次使用ax.scatter()，这必须是这样（例如，我使用不同的颜色）。

问候

Answer 1

您可以使用 event.artist 的_offsets3d属性来获取坐标数据，然后使用 ind 获取选取的点：

def onpick(event):
    ind = event.ind[0]
    x, y, z = event.artist._offsets3d
    print x[ind], y[ind], z[ind]