如何将数学ODE代码更改为Python

时间:2019-07-04 10:48:53

标签: python wolfram-mathematica ode odeint

Mathematica内给出了以下代码

solution = NDSolve[{\[CapitalOmega]\[CapitalLambda]'[x] == 
(1 - \[CapitalOmega]\[CapitalLambda][x]) \[CapitalOmega]\[CapitalLambda][x] ((1 - \[CapitalOmega]\[CapitalLambda][x])^((1 - \[Delta])/(4 -\2 \[Delta]))  \[CapitalOmega]\[CapitalLambda][x]^(1/(4 - 2 \[Delta]))), \[CapitalOmega]\[CapitalLambda][0] == \[CapitalOmega]\[CapitalLambda]f}, \[CapitalOmega]\\[CapitalLambda], {x, xi, xf}];

我想用ODEINT用Python编写 但是我真的不明白它的编写方式,因为它会发送`error1复数。

def OD_H(z, od, delt):
    dMdt = od * (1 - od) * ((1 - od)**((1 - delt)/(4-2*delt)) * od**(1/(2 *(2-delt))))
    return dMdt                                                                                                                      

def ant(z, od0, delt):
    z1 = 0
    od = odeint(OD_H, od0, [z1, z], args=(delt,))[-1]       
    return od     

for z in np.arange(0,3.1,0.1):
    print(ant(z, 0.7, 1.1))

错误是

od = odeint(OD_H, od0, [z1, z], args=(delt,))[-1] File "C:\Python36-32\lib\site-packages\scipy\integrate\odepack.py", line 244, in odeint int(bool(tfirst))) TypeError: can't convert complex to float

1 个答案:

答案 0 :(得分:1)

通过odeint调用,我们了解到您将z用作时间变量并将od用作空间/状态变量。由于您的派生函数具有时间优先,并且odeint中的默认值是具有状态优先,因此您需要设置选项tfirst=True。有了这一更改,就不会出错。通过使用odeint的全部功效(列出所有所需采样点的列表),您可以用更少的计算获得相同的值

z = np.arange(0,3.1,0.1)
od = odeint(OD_H, od0, z, args=(delt,), tfirst=True)

然后z[k], od[k]对与您在循环中计算的参数值对相对应。