odeint Python 的问题

Question

我正在尝试使用 odeint，但我有一个参数会随着每个时间步的变化而改变值。我的功能是-

def f(y, t, k1, k2, Pr):

    return k1*(Pr-k2)*y

其中Pr 是一个包含浮点值的熊猫系列，我观察到了其中的值。 anft是从1961年到2014年的整数年列表。我为odeint写了以下代码-

y0 = 120000
k1 = 0.2
k2 = -.4
y = odeint(f, y0, t, args = (k1, k2, Pr, ))

但它给出了以下错误-

RuntimeError: The size of the array returned by func (54) does not match the size of y0 (1).

Answer 1

如错误消息所示，Pr 作为长度为 54 的向量传入，而不是一个接一个的值。推断Pr 是一个NumPy 向量，这意味着f 的输出是将return 值元素应用于Pr 的结果。

要解决此问题，您可以改用从时间输入生成Pr 值的函数。再次假设Pr[k]对应t[k]，这个函数的最简单版本可能是：

def get_Pr(time: float):
    t_index = sum(t <= time) - 1 
    return Pr[t_index]

然后修改f如下：

def f(y, t, k1, k2):
    this_pr = get_Pr(t)
    return k1*(this_pr-k2)*y

编辑：...然后不要将Pr 作为参数传递给scipy.odeint，因为f 以不同的方式获取值。