使用scipy.optimize 的curve_fit函数对观测值拟合

Posted 2020-12-14 jiaxinwei

x=np.arange(1,13,1)
y=np.array([17, 19, 21, 28, 33, 38, 37, 37, 31, 23, 19, 18 ])
plt.plot(x,y)
plt.show()

可以看出温度是以周期为12的正弦函数

#构建函数y=a*sin(x*pi/6+b)+c
#使用optimize.curve_fit函数求出a、b、c的值

x=np.arange(1,13,1)
x1=np.arange(1,13,0.1)
ymax=np.array([17, 19, 21, 28, 33, 38, 37, 37, 31, 23, 19, 18 ])

def fmax(x,a,b,c):
    return a*np.sin(x*np.pi/6+b)+c

fita,fitb=optimize.curve_fit(fmax,x,ymax,[1,1,1]) #[1,1,1]是初始化的参数
print(fita) #参数
print(fitb) #参数的协方差矩阵
plt.plot(x,ymax)
plt.plot(x1,fmax(x1,fita[0],fita[1],fita[2]))
plt.show()