根据某地每月的平均温度[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]
以上为拟合曲线