所以我找到了这个代码:https : //matplotlib.org/examples/event_handling/legend_picking.html。我试图让它适用于数据点而不是线。所以我以为我只是更改了“o”中的标记,但这似乎不起作用。我还想在动画中使用它,以便我可以决定是否要跟踪数据点。最后我想要 10 个可点击的图例条目。
我要做的是在第 1 行和第 2 行代码中放置一个标记:
import numpy as np
import matplotlib.pyplot as plt
t = np.arange(0.0, 0.2, 0.1)
y1 = 2*np.sin(2*np.pi*t)
y2 = 4*np.sin(2*np.pi*2*t)
fig, ax = plt.subplots()
ax.set_title('Click on legend line to toggle line on/off')
line1, = ax.plot(t, y1, ls='None', marker='o', color='red', label='1 HZ')
line2, = ax.plot(t, y2, ls='None', marker='o', color='blue', label='2 HZ')
leg = ax.legend(loc='upper left', fancybox=True, shadow=True)
leg.get_frame().set_alpha(0.4)
# we will set up a dict mapping legend line to orig …Run Code Online (Sandbox Code Playgroud)