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

Question

我使用mplo3d绘制一系列点:

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]