如何在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()，而且必须这样做（例如，我使用不同的颜色）。

谢谢

- user1371437

Answer 1

您可以使用事件对象的 _offsets3d 属性获取坐标数据，然后使用ind获取选中的点：

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

- HYRY