matplotlib实现交换式图形显示

一 代码

1. from random import choice
2. import numpy as np
3. import matplotlib.pyplot as plt
4. from matplotlib.widgets importRadioButtons,Button
6. t = np.arange(0.0,2.0,0.01)
7. s0 = np.sin(2*np.pi*t)
8. s1 = np.sin(4*np.pi*t)
9. s2 = np.sin(8*np.pi*t)
11. fig, ax = plt.subplots()
12. l,= ax.plot(t, s0, lw=2, color='red')
13. plt.subplots_adjust(left=0.3)
15. #定义允许的几种频率，并创建单选钮组件
16. #其中[0.05, 0.7, 0.15, 0.15]表示组件在窗口上的归一化位置
17. axcolor ='lightgoldenrodyellow'
18. rax = plt.axes([0.05,0.7,0.15,0.15], axisbg=axcolor)
19. radio =RadioButtons(rax,('2 Hz','4 Hz','8 Hz'))
20. hzdict ={'2 Hz': s0,'4 Hz': s1,'8 Hz': s2}
21. def hzfunc(label):
22. ydata = hzdict[label]
23. l.set_ydata(ydata)
24. plt.draw()
25. radio.on_clicked(hzfunc)
27. #定义允许的几种颜色，并创建单选钮组件
28. rax = plt.axes([0.05,0.4,0.15,0.15], axisbg=axcolor)
29. colors =('red','blue','green')
30. radio2 =RadioButtons(rax, colors)
31. def colorfunc(label):
32. l.set_color(label)
33. plt.draw()
34. radio2.on_clicked(colorfunc)
36. #定义允许的几种线型，并创建单选钮组件
37. rax = plt.axes([0.05,0.1,0.15,0.15], axisbg=axcolor)
38. styles =('-','--','-.','steps',':')
39. radio3 =RadioButtons(rax, styles)
40. def stylefunc(label):
41. l.set_linestyle(label)
42. plt.draw()
43. radio3.on_clicked(stylefunc)
45. #定义按钮单击事件处理函数，并在窗口上创建按钮
46. def randomFig(event):
47. #随机选择一个频率，同时设置单选钮的选中项
48. hz = choice(tuple(hzdict.keys()))
49. hzLabels =[label.get_text()for label in radio.labels]
50. radio.set_active(hzLabels.index(hz))
51. l.set_ydata(hzdict[hz])
52. #随机选择一个颜色，同时设置单选钮的选中项
53. c = choice(colors)
54. radio2.set_active(colors.index(c))
55. l.set_color(c)
56. #随机选择一个线型，同时设置单选钮的选中项
57. style = choice(styles)
58. radio3.set_active(styles.index(style))
59. l.set_linestyle(style)
60. #根据设置的属性绘制图形
61. plt.draw()
62. axRnd = plt.axes([0.5,0.015,0.2,0.045])
63. buttonRnd =Button(axRnd,'Random Figure')
64. buttonRnd.on_clicked(randomFig)
65. #显示图形
66. plt.show()

二 运行结果