wxPython的matplotlib - 检测被点击的3D线图

Question

哪一行我使用this example code作为我的起始代码

此代码生成的柱状图。当用户点击条形图上的实际条形时，将触发事件并显示单个条形的边框坐标。这发生在on_pick函数中。

我修改了代码，以代替条形图，我有一个Axes3D line plot。我在这个XYZ坐标系上绘制了几条单独的线。我需要的是识别哪一行被点击。看起来，边界框不适用于三维绘图，因为它与条形图一样。

我已经查看了pick_event文档，但一直未能找到解决方案。我可以简单地检测点击位置的x，y坐标，但是因为可以以任意角度旋转绘图，所以检测哪条线实际被点击是非常困难的，尤其是因为它们可以根据视点重叠。

我感谢您的时间和帮助！

Answer 1

如果您只需要知道哪个曲线被点击，这不是很困难。您需要创建一个line_picker（）来检查鼠标位置是否足够接近曲线。

line_picker（）需要计算点和线段之间的距离，这有点困难，我使用曲线的线性插值来解决这个问题。

这里是代码，我插入2000点的曲线，如果鼠标位置的曲线小于5像素，然后曲线被选中。

import matplotlib as mpl 
from mpl_toolkits.mplot3d import Axes3D 
import numpy as np 
import matplotlib.pyplot as plt 

def line_picker(line, me): 
    if me.xdata is None: return False, dict()  
    x, y = me.x, me.y 
    xdata, ydata = line.axes.transData.transform(np.array(line.get_data()).T).T 
    index = np.arange(len(xdata)) 
    index2 = np.linspace(0, index[-1], 2000) 
    xdata2 = np.interp(index2, index, xdata) 
    ydata2 = np.interp(index2, index, ydata) 
    d = np.sqrt((xdata2-x)**2. + (ydata2-y)**2.) 
    if np.min(d) < 5: 
     return True, {} 
    else: 
     return False, {} 

mpl.rcParams['legend.fontsize'] = 10 

fig = plt.figure() 
ax = fig.gca(projection='3d') 
theta = np.linspace(-4 * np.pi, 4 * np.pi, 100) 
z = np.linspace(-2, 2, 100) 
r = z**2 + 1 
x = r * np.sin(theta) 
y = r * np.cos(theta) 
line1 = ax.plot(x, y, z, label='parametric curve', picker=line_picker)[0] 

t = np.linspace(-1, 1, 100) 
x = 4*np.sin(10*t) 
y = 4*np.cos(10*t) 
z = t**2*5-3 

line2 = ax.plot(x, y, z, label="second", picker=line_picker)[0] 

ax.legend() 

def onpick(event): 
    print [line1, line2].index(event.artist) 

fig.canvas.mpl_connect('pick_event', onpick) 

plt.show()