我正在尝试拟合一些数据,这些数据开始平坦,在时间 = 0 处急剧达到峰值,然后衰减为两个负指数(首先作为快速指数,然后在一小段时间后作为较慢指数)。我一直在尝试使用 scipy.optimize 中的curve_fit,但拟合似乎无法识别第一个快速衰减的指数。
我所做的只是为不同的时间范围定义两个拟合函数。对于 time<0 我有一个常数(我称之为 line 因为起初我使用的是一条线)。对于时间> 0,我定义了具有不同参数的两个指数的总和。然后我对这些参数进行猜测,并将所有内容输入curve_fit。
我真的只是想知道是否有人对如何让它识别第一个峰值和快速衰减有任何想法......
在我的谷歌搜索中,我只能找到拟合简单数据的示例(例如单个指数或多项式)。我可能完全错误地接近这个,任何帮助将不胜感激。谢谢!
另外,我会附上我得到的东西,但这是我的第一篇文章,我没有声誉,所以它不会让我......
这是数据的链接:http: //www.lehigh.edu/~ivb2/teaching/smp/current/DecayData.txt
这是代码:
#!/usr/bin/env python
import numpy as np
from scipy import optimize
from scipy.optimize import curve_fit
import matplotlib.pyplot as plt
def line(x, *p):
return p[0]
def exponential(x, *p):
return p[0] * np.exp(p[1] * x) + p[2]
#Fit function (neg exp in this case)
def fitfunc(time, *p):
tempArray = np.empty_like(time)
tempArray[time <= 0] = line(time, p[6])
tempArray[0 < time] = exponential(time, p[0], p[1], p[2]) + exponential(time, p[3], p[4], p[5])
return tempArray
#Read in the data to 3 arrays
time, decay1, decay2 = np.loadtxt('DecayData.txt', unpack=True)
#An initial guess for the fit parameters
guess = np.array([5e-2, -2500, 0, 5e-2, -2000, 0, 0])
#Fits the functions using curve_fit
popt, pcov = curve_fit(fitfunc, time, decay1, guess)
print popt
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(time, decay1)
ax.plot(time, fitfunc(time, *popt))
plt.show()