图例选择

启用图例上的拾取以打开和关闭原始线。

图例选择

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  4. t = np.arange(0.0, 0.2, 0.1)
  5. y1 = 2*np.sin(2*np.pi*t)
  6. y2 = 4*np.sin(2*np.pi*2*t)
  8. fig, ax = plt.subplots()
  9. ax.set_title('Click on legend line to toggle line on/off')
  10. line1, = ax.plot(t, y1, lw=2, color='red', label='1 HZ')
  11. line2, = ax.plot(t, y2, lw=2, color='blue', label='2 HZ')
  12. leg = ax.legend(loc='upper left', fancybox=True, shadow=True)
  13. leg.get_frame().set_alpha(0.4)
  16. # we will set up a dict mapping legend line to orig line, and enable
  17. # picking on the legend line
  18. lines = [line1, line2]
  19. lined = dict()
  20. for legline, origline in zip(leg.get_lines(), lines):
  21.     legline.set_picker(5)  # 5 pts tolerance
  22.     lined[legline] = origline
  25. def onpick(event):
  26.     # on the pick event, find the orig line corresponding to the
  27.     # legend proxy line, and toggle the visibility
  28.     legline = event.artist
  29.     origline = lined[legline]
  30.     vis = not origline.get_visible()
  31.     origline.set_visible(vis)
  32.     # Change the alpha on the line in the legend so we can see what lines
  33.     # have been toggled
  34.     if vis:
  35.         legline.set_alpha(1.0)
  36.     else:
  37.         legline.set_alpha(0.2)
  38.     fig.canvas.draw()
  40. fig.canvas.mpl_connect('pick_event', onpick)
  42. plt.show()

下载这个示例