我想为时间序列数据拟合一个定义如下的函数:
def func(t, a0, a1, a2, T, tau1, tau2):
if t < T:
return a0 + a1 * np.exp(-t/tau1) + a2 * np.exp(-t/tau2)
else:
return a0 + a1 * np.exp(-T/tau1) * (1 - t/tau1 + T/tau1) + a2 * np.exp(-T/tau2) * (1 - t/tau2 + T/tau2)
Run Code Online (Sandbox Code Playgroud)
这里,t表示进行测量的时间,其余参数是函数的参数.问题在于,当我将它提供给curve_fit时,Python抱怨t <T比较中的模糊性.我相信这是因为当在curve_fit中调用func时t变为数据点列表,而T是数字(不是列表):
popt, pcov = curve_fit(func, t1, d1)
Run Code Online (Sandbox Code Playgroud)
其中t1是时间列表,d1是在相应时间测量的数据值列表.我尝试过多种方法来解决这个问题,但无济于事.有什么建议吗?非常感谢!