问题描述
x_data = [3,6,9,12,16,21,28,50]
和
y_data = [0,333.33333333,250,200,142.85714286,45.45454545]
我附上了这张图:
我需要将其拟合到具有函数的曲线:f(x) = a * exp(-x / b) + c。
但是我无法让 scipy curve_fit
工作。我该怎么办?
解决方法
您必须定义要拟合的函数,然后将其传递给 curve_fit
。您的函数必须首先采用自变量,然后采用所有其余参数。
def func(x,a,b,c):
return a * np.exp(-x / b) + c
# Get parameters and covariance
params,cov = scipy.optimize.curve_fit(f=f,xdata=x_data,ydata=y_data)
# Declare dedicated letters for each parameter
a,c = params
# Plot
plt.plot(x_data,y_data)
plt.plot(x_data,[func(x,a=a,b=b,c=c) for x in x_data])
plt.show()