使用scipy ode解算器时的尺寸值错误

Question

当前尝试运行以下代码

import numpy as np

from scipy.interpolate import interp1d

from scipy.integrate import ode


def dydt(t,y,tsi,rho):

    Lambda      = 10^-4
    beta        = 0.0065
    lambda2      = 0.08

    rho = interp1d(tsi,rho, fill_value = 'extrapolate')


    #one group delayed precursor
    dydt1 = (-lambda2*y[0] + beta*y[1])
    #power
    dydt2 = (((rho(t)-beta)/Lambda)*y[1]+(lambda2*y[0])/Lambda)
    return dydt1, dydt2


x = 21

dt = 1

tsi=np.arange(0,x,dt)
dt = [1]

rho=np.ones(x)*0.0025
y0= [1,0]
t0 = [0,x-1]


sol = []

i = ode(dydt)

i.set_integrator('dopri5')

i.set_initial_value(t0,y0)

i.set_f_params(tsi,rho)


for t in tsi[1:]:

    i.integrate(i.t+dt)

    sol.append(i.y)

但是，我最终犯了一个错误：

runfile（'C:/Users/-/Desktop/-.py'，wdir='C:/Users/-/Desktop'）回溯（最后一次调用）：

文件“C:\Users---\Desktop---.py”，第46行，在i.integrate（i.t dt）中

文件“C:\Users---\anaconda3\lib\site packages\scipy\integrate\u ode.py”，第432行，在integrate self中。f_参数，赛尔夫。雅克（美国）

文件“C:\Users---\anaconda3\lib\site packages\scipy\integrate_ode.py”，第1172行，运行元组（self.call_args）（f_参数，））

ValueError:第0个维度必须为2，但得到0（未定义）。

我在网上遵循了一个类似的例子，但主要的区别是，我在函数中返回2个方程，我的两个参数是数组。我试图使用odeint和solve_ivp，但他们给出了非常不准确的答案。

非常感谢。

Answer 1

再次检查线路

i.set_initial_value(t0,y0)

在文档中，时间参数排在第二位（因为它是一个默认值为0的可选参数）。初始时间是一个标量，而不是元组。

您应该只定义一次内插函数，然后只将其作为参数传递，甚至将其作为全局对象。

sol.append(i.y)

也可能不做你认为它做的事情，在向量值的情况下，它构建了一个引用的相同副本的数组i. y。使用i.y.copy（）生成从求解器内部删除的副本。