使用Slider组件调整曲线参数

一 代码

1. import numpy as np
2. import matplotlib.pyplot as plt
3. from matplotlib.widgets importSlider,Button,RadioButtons
5. fig, ax = plt.subplots()
6. plt.subplots_adjust(left=0.1, bottom=0.25)
7. t = np.arange(0.0,1.0,0.001)
8. #初始振幅与频率，并绘制初始图形
9. a0, f0=5,3
10. s = a0*np.sin(2*np.pi*f0*t)
11. l,= plt.plot(t, s, lw=2, color='red')
12. #设置坐标轴刻度范围
13. plt.axis([0,1,-10,10])
15. axColor ='lightgoldenrodyellow'
16. #创建两个Slider组件，分别设置位置/尺寸、背景色和初始值
17. axfreq = plt.axes([0.1,0.1,0.75,0.03], axisbg=axColor)
18. sfreq =Slider(axfreq,'Freq',0.1,30.0, valinit=f0)
19. axamp = plt.axes([0.1,0.15,0.75,0.03], axisbg=axColor)
20. samp =Slider(axamp,'Amp',0.1,10.0, valinit=a0)
21. #为Slider组件设置事件处理函数
22. def update(event):
23. #获取Slider组件的当前值，并以此来更新图形
24. amp = samp.val
25. freq = sfreq.val
26. l.set_ydata(amp*np.sin(2*np.pi*freq*t))
27. plt.draw()
28. #fig.canvas.draw_idle()
29. sfreq.on_changed(update)
30. samp.on_changed(update)
32. def adjustSliderValue(event):
33. ampValue = samp.val +0.05
34. if ampValue >10:
35. ampValue =0.1
36. samp.set_val(ampValue)
38. freqValue = sfreq.val +0.05
39. if freqValue >30:
40. freqValue =0.1
41. sfreq.set_val(freqValue)
42. update(event)
43. axAdjust = plt.axes([0.6,0.025,0.1,0.04])
44. buttonAdjust =Button(axAdjust,'Adjust', color=axColor, hovercolor='red')
45. buttonAdjust.on_clicked(adjustSliderValue)
47. #创建按钮组件，用来恢复初始值
48. resetax = plt.axes([0.8,0.025,0.1,0.04])
49. button =Button(resetax,'Reset', color=axColor, hovercolor='yellow')
50. def reset(event):
51. sfreq.reset()
52. samp.reset()
53. button.on_clicked(reset)
55. ###用来控制图形颜色的
56. ##rax = plt.axes([0.025, 0.5, 0.15, 0.15], axisbg=axColor)
57. ##radio = RadioButtons(rax, ('red', 'blue', 'green'), active=0)
58. ##def colorfunc(label):
59. ## l.set_color(label)
60. ## fig.canvas.draw_idle()
61. ##radio.on_clicked(colorfunc)
63. plt.show()

二 运行结果