利用Python进行多项式拟合

多项式拟合的简单代码:

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()

结果: