找到指数曲线拟合的初始猜测

时间:2017-07-26 21:42:39

标签: python numpy curve-fitting exponential-distribution

所以我不知道怎么做,而且我不确定是否有一些我现在完全忘记的数学但我不知所措。简而言之,我有一些使用scipy和numpy的相当简单的初始代码,我希望它符合指数曲线:

from scipy.optimize import curve_fit
import numpy as np

# sample data
x = np.array([7620.,   7730.,   7901.,   8139.,   8370.,   8448.,   8737.,   8824., 9089.,   9233.,   9321.,   9509.,   9568.,   9642.,   9756.,   9915.,  10601., 10942.])
y = np.array([0.01228478,  0.01280466,  0.01363498,  0.01493918,  0.01530108, 0.01569484,  0.01628133,  0.01547824,  0.0171548,   0.01743745,  0.01776848, 0.01773898,  0.01839569,  0.01823377,  0.01843686,  0.01875542,  0.01881426, 0.01977975])

# define type of function to search
def model_func(x, a, k, b):
    return a * np.exp(-k*x) + b

# curve fit
p0 = (2.e-6,300,1)
opt, pcov = curve_fit(model_func, x, y, p0)
a, k, b = opt
# test result
x2 = np.linspace(7228, 11000, 3000)
y2 = model_func(x2, a, k, b)
fig, ax = plt.subplots()
ax.plot(x2, y2, color='r', label='Fit. func: $f(x) = %.3f e^{%.3f x} %+.3f$' % (a,k,b))
ax.plot(x, y, 'bo', label='data with noise')
ax.legend(loc='best')
plt.show()

我的问题是尝试,因为我可能无法弄清楚p0的初始参数 - 我尝试了一系列值,但坦率地说我不知道​​我在做什么所以我没有得到解决方案这里。有人可以建议怎么做吗?谢谢!

1 个答案:

答案 0 :(得分:0)

对我来说,这个数据更接近于指数而不是指数曲线,你确定你的模型是对的吗?

关于最初的猜测,我假设您对该功能没有任何进一步的了解,如果是这样的话,请使用它:

对于x-> \ inf,fucntion接近b。所以我会用b估计大约0.025。 对于其他两个变量,采用xy的子样本并明确解释等式:

a * np.exp(-k*x[0]) + 0.025 = y[0]
a * np.exp(-k*x[-1]) + 0.025 = y[-1]

解决这个问题:

a = (y[0]-0.025)/np.exp(-k*x[0])
e^(-k*(x[-1]-x[0])=(y[-1]-0.025)/(y[0]-0.025) # and then take logarithm

k = -np.log((y[-1]-0.025)/(y[0]-0.025))/(x[-1]-x[0])
a = (y[0]-0.025)/np.exp(-k*x[0])

提供k=0.00026798747972760543a=-0.80114087848462689

这种方法可以一般使用,扔掉尽可能多的点来解决精确的方程,然后用这些值来启动最优值