使用 SymPy 表达式和 SciPy 求解器求解一阶 ODE 系统

Question

我正在尝试学习如何将 python SymPy 与 SciPy 相结合以数值求解常微分方程。但是，我有点迷失了如何将 ODE 的一阶系统的 SymPy 形式实际转换为我可以使用 scipy.integrate.odeint()

处理的格式

请注意，有些人认为这与另一个相似 post，但事实并非如此。另一个 post 在这里。
Convert sympy expressions to function of numpy arrays

所以另一个 post 是一个更复杂的情况，用户想要使用 Theano 或其他一些库来加速 ODE 的计算。我只是想了解 SymPy 和 SciPy 之间的基本接口，所以这个 post 一点帮助也没有。

作为玩具示例，我使用 Lotka-Volterra 方程来测试 SymPy 的使用。等式是：

$\dot{x} = ax - bxy$

$\dot{y} = cxy - dy$

我可以用 Scipy 以常规方式解决这个问题，而且它有效。这是工作代码。

import numpy as np
import scipy
from scipy.integrate import odeint, ode, solve_ivp
import sympy
import matplotlib.pyplot as plt
sympy.init_printing()

def F_direct(X, t, args):
    F = np.zeros(2)
    a, b, c, d = args
    x,y = X
    F[0] = a*x - b*x*y
    F[1] = c*x*y- d*y
    return F

argst = [0.4,0.002,0.001,0.7]
xy0 = [600, 400]
t = np.linspace(0, 50, 250)
xy_t, infodict = odeint(F_direct, xy0, t, args=(argst,), full_output=True)

plt.plot(t, xy_t[:,1], 'o', t, xy_t[:,0])
plt.grid(True)
plt.xlabel('x'); plt.ylabel('y')
plt.legend(('Numerical', 'Exact'), loc=0)
plt.show()

现在我对如何使用 SymPy 做到这一点感到迷茫。我知道需要做什么，但不确定如何进行。我发现的唯一示例太复杂而无法学习。这是我的。

x, y, a, b, c, d = sympy.symbols('x y a b c d')
t = np.linspace(0, 50, 250)
ode1 = sympy.Eq(a*x - b*x*y)
ode2 = sympy.Eq(c*x*y - d*y)

我应该将这些方程式放入某种系统形式，然后使用 sympy.lambdify 函数来 return 我可以传递给 odeint[=20 的新函数=]

因此，任何人都可以在此处填写有关我如何设置此 ode1,ode2 系统以使用 SymPy 进行处理的空白。

Answer 1

很少需要在SymPy中使用Eq对象；方程最好用表达式表示，在求解器的上下文中，表达式被理解为等于零。因此，ode1 和 ode2 应该只是 ODE 系统 right-hand 端的表达式。然后可以一起lambdified，如下：

ode1 = a*x - b*x*y
ode2 = c*x*y - d*y
F_sympy = sympy.lambdify((x, y, a, b, c, d), [ode1, ode2])
def F_from_sympy(X, t, args):
    a, b, c, d = args
    x, y = X    
    return F_sympy(x, y, a, b, c, d)

需要 lambdification 之后的额外步骤，因为 SciPy 的求解器传递了一些数组，lambdified 函数不知道如何解包。示例：

F_from_sympy([1, 2], np.linspace(0, 1, 100), (3, 4, 5, 6))

returns [-5, -2] 这是一个 Python 列表而不是 NumPy 数组，但 ODE 求解器应该处理它。（或者你可以 return np.array(F_sympy(x, y, a, b, c, d)))。）

Answer 2

我写了 a module named JiTCODE，它从描述 right-hand 端的符号表达式（SymPy 或 SymEngine）创建 ODE 积分器对象（处理方式类似于 scipy.integrate.ode）。在引擎盖下，它使用 scipy.integrate.ode 和 scipy.integrate.solve_ivp 进行集成。

你的情况的唯一缺点是动态变量和时间的符号是规定的，因此你可能必须进行符号替换——但这应该不是什么大问题。下面，以你的 Lotka–Volterra 方程为例，使用 y(0) 代替 x 和 y(1) 代替 y.

import numpy as np
from sympy.abc import a,b,c,d
from jitcode import y, jitcode

xy0 = [600,400]
argst = [0.4,0.002,0.001,0.7]

lotka_volterra = [
         a*y(0) - b*y(0)*y(1),
        -d*y(1) + c*y(0)*y(1)
    ]

ODE = jitcode( lotka_volterra, control_pars=[a,b,c,d] )
ODE.set_integrator("dopri5")
ODE.set_initial_value(xy0)
ODE.set_parameters(*argst)

times = np.linspace(0, 50, 250)
xy_t = np.vstack(ODE.integrate(time) for time in times)

请注意，此模块的主要特点是 right-hand 端是为了提高效率而编译的。根据您需要执行的操作，这可能有点矫枉过正，但如果有效也无妨（您也可以禁用它并使用中详述的 lambdification）。

使用 SymPy 表达式和 SciPy 求解器求解一阶 ODE 系统

Solving a first order system of ODEs using SymPy expressions and SciPy solver

python

sympy

scipy

ode