如何在 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]