将参数和初始值的字典传递给scipy.integrate.odeint

时间:2017-05-18 16:24:15

标签: dictionary parameters scipy integrate odeint

我正在尝试使用spicy.itegrate.odeint集成一个微分方程组。

首先,对参数和初始条件进行采样并以两个字典(x0p)返回。然后创建模型并将其作为函数写入文件,大致如下(使用虚拟方程式):

def model(x, t, p):
    xdot = [
        x['rate1'], p["a"]
        x['rate2'], p["b"] * x["state1"] - p["c"] * x["state2"]
        x['rate3'], p["c"] * x["state2"]
        x["state4"], x["rate1"] + x["rate2"] 
        x["state5"], - x["rate2"] + x["rate3"]
        ]
    return xdot

这样我就可以从简单的输入中轻松生成不同的模型。因此,通常可能是硬编码的变量,现在是具有相应值的字典中的键。我这样做是因为动态分配变量被认为是不好的做法。

当我尝试使用odeint集成系统时,如下

sol = odeint(model, x0, t, args=(p,),
              atol=1.0e-8, rtol=1.0e-6)

因此,x0是初始条件的字典,p参数(和t浮点列表)。我收到以下错误:

TypeError: float() argument must be a string or a number, not 'dict'

显然,scipy对我尝试传递字典以参数化和初始化我的模型感到不满意。问题是我是否有办法解决这个问题,或者我是否被迫将字典中的所有值分配给具有相应键名称的变量。后者不允许我将相同的初始条件和参数集传递给所有模型,因为它们在状态和参数方面都不同。因此,希望将相同的参数集传递给所有模型,无论参数在模型中是否存在。

1 个答案:

答案 0 :(得分:2)

出于性能原因,像odeint这样的scipy函数可以处理数组,其中每个参数都与固定位置相关联。

按名称访问参数的解决方案是将它们转换为namedtuple,并为它们提供名称和位置。但是,转换需要在函数内部完成,因为odeint将参数作为numpy.array传递给模型函数。

这个例子应该传达这个想法:

from scipy.integrate import odeint
from collections import namedtuple

params = namedtuple('params', ['a', 'b', 'c', 'd'])

def model(x, t0):
    x = params(*x)
    xdot = [1, 
            x.a + x.b, 
            x.c / x.a, 
            1/x.d**2]  # whatever
    return xdot


x0 = params(a=1, b=0, c=2, d=0.5)

t = [0, 0.5, 1]

sol = odeint(model, x0, t)