使用scipy.integrate.odeint与字典[Python]

Question

我试图解决使用scipy.integrate.odeint衰变方程。我试图从字典中获得初始值，但它不起作用，我不确定它是否可以工作。这里是我正在使用的代码：

from scipy.integrate import odeint 
import numpy as np 
import matplotlib.pyplot as plt 

def decay(init,t): 
    f0 = - init['a']/.5 
    f1 = init['a']/.5 - init['b']/.2 
    f2 = init['b']/.2 
    return [f0,f1,f2] 

if __name__ == '__main__': 
    init = {'a':5, 'b':0, 'c':0} 
    time = np.linspace(0, 10, 101) 
    soln = odeint(decay, init ,time) 
    a = soln[:,0] 
    b = soln[:,1] 
    c = soln[:,2] 
    print a 
    print b 
    print c 
    plt.plot(time, a, color = 'g') 
    plt.plot(time, b, color = 'r') 
    plt.plot(time, c, color = 'b') 
    plt.show() 

它的工作原理是，如果预期的，而不是一本字典我用像这样的列表：

from scipy.integrate import odeint 
import numpy as np 
import matplotlib.pyplot as plt 

def decay(init,t): 
    a,b,c = init 
    f0 = - a/.5 
    f1 = a/.5 - b/.2 
    f2 = b/.2 
    return [f0,f1,f2] 

if __name__ == '__main__': 
    init = [5,0,0] 
    time = np.linspace(0, 10, 101) 
    soln = odeint(decay, init ,time) 
    a = soln[:,0] 
    b = soln[:,1] 
    c = soln[:,2] 
    print a 
    print b 
    print c 
    plt.plot(time, a, color = 'g') 
    plt.plot(time, b, color = 'r') 
    plt.plot(time, c, color = 'b') 
    plt.show() 

不过，我需要使用一个字典我的目的。有没有办法使用字典来调用初始值？

Answer 1

如果这个工程：

init = [5,0,0] 
time = np.linspace(0, 10, 101) 
soln = odeint(decay, init ,time) 

那么这应该还有：

adict = {'a':5, 'b':0, 'c':0} 
init = [adict['a'],adict['b'],adict['c']] 
time = np.linspace(0, 10, 101) 
soln = odeint(decay, init ,time) 

换句话说，无论你在哪里得到这个字典中，你需要将其值转换成名单。

init = adict.values()（或PY3 list(adict.values())）将行不通，因为字典订单我行我素键：

In [306]: list({'a':5, 'b':0, 'c':0}.values()) 
Out[306]: [0, 0, 5] 

或键列表较长，这可能是简单的：

In [307]: adict = {'a':5, 'b':0, 'c':0} 
In [308]: init = [adict[k] for k in ['a','b','c']] 
In [309]: init 
Out[309]: [5, 0, 0]