Python 曲线拟合

 1 #曲线拟合
 2 fig = plt.figure()
 3 ax = fig.add_subplot(111)#将画布分割成1行1列,图像画在从左到右从上到下的第1块
 4 
 5 ax.plot(Num,a,label=u'A',color='m',linestyle='',marker='.')
 6 ax.plot(Num,b,label=u'B',color='g',linestyle='',marker='.')
 7 #计算多项式
 8 c=np.polyfit(Num,a,100)#拟合多项式的系数存储在数组c中
 9 yy=np.polyval(c,Num)#根据多项式求函数值
10 c2=np.polyfit(Num,b,100)#拟合多项式的系数存储在数组c中
11 yy2=np.polyval(c2,Num)#根据多项式求函数值
12 #进行曲线绘制
13 x_new=np.linspace(0, 360)
14 f_liner=np.polyval(c,x_new)
15 f_liner2=np.polyval(c2,x_new) 
18 ax.plot(x_new,f_liner,label=u'A',color='g',linestyle='-',marker='')
19 ax.plot(x_new,f_liner2,label=u'B',color='b',linestyle='-',marker='')
20 # labels标签设置
21 ax.set_xlim(0, 366)
22 ax.set_xlabel(u'T')
23 ax.set_ylabel(u'Values of A/B')
24 ax.set_title(u'A/B By T', bbox={'facecolor':'0.8', 'pad':5})
25 ax.legend()
26 plt.show()
27 #

原文地址:https://www.cnblogs.com/Ann21/p/9487920.html