python实现数据拟合

主要用到了numpy的 polyfit与poly1d函数

import numpy as np
import matplotlib.pyplot as plt
x,y = [1,1.25,1.33,1.60,2.0,2.4,2.5,3.0,3.2],[1000,1100,1300,1500,1861.48\
      ,2233.6,2326.64,2791.94,2978.09]

z1=np.polyfit(x, y, 3)#3表示3阶拟合
p1=np.poly1d(z1)#p1就是xy的关系式
print(p1)#打印x，y的关系表达式
y_fit=p1(x)#使用这个关系式生成拟合数据
plt.scatter(x,y_fit,label='fit')#画拟合图
plt.scatter(x,y,label='data')#画原始图
plt.legend()
plt.show()

3阶拟合的结果如下：