python scipy学习-曲线拟合

根据某地每月的平均温度[17, 19, 21, 28, 33, 38, 37, 37, 31, 23, 19, 18]拟合温度函数。

import numpy as np
import matplotlib.pyplot as plt
from scipy import optimize

scipy.optimize提供了函数最小值(标量或多维)、曲线拟合和寻找等式的根的有用算法。

因为温度是以年为单位的

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])
print(fita)
plt.plot(x,ymax)
plt.plot(x1,fmax(x1,fita[0],fata[1],fita[2]))
plt.show()

求出a b c的值为[ 10.93254952  -1.9496096   26.75]

以上为拟合曲线