多项式拟合的简单代码:
import matplotlib.pyplot as plt import numpy as np x=[1,2,3,4,5,6,7,8] y=[1,4,9,13,30,25,49,70] a=np.polyfit(x,y,2) #用2次多项式拟合x,y数组 b=np.poly1d(a) #拟合完成后生成多项式对象 c=b(x) #获取x在多项式处的值 plt.scatter(x,y,marker='o',label='original datas') #对原始数据做散点图 plt.plot(x,c,ls='--',c='red',label='fitting with second polynomial')#对拟合之后的数据作图 plt.legend() #给图加上图例 plt.show()
结果:
