python - Sympy - ufuncify 向量函数

Question

我对修改后的FitzHugh-Nagumo 模型的向量函数进行了lambdify-ied （没有扩散项）：

from sympy import symbols, Matrix, solve, Eq
from sympy import diff, simplify, lambdify
from sympy.utilities.autowrap import ufuncify


u, v , e, a0, a1 = symbols('u v e a0 a1')

dudt = u - u**3 -v
dvdt = e*(u - a1*v - a0)

eqs = Matrix([dudt,dvdt])
print eqs

numerical_eqs = eqs.subs([(e,1.0),(a0,0.5),(a1,1.0)])
print numerical_eqs
lambdify_eqs = lambdify([u,v], numerical_eqs)
print lambdify_eqs(1.0,0.5)

但是尝试以ufunc相同的方式生成 a 会导致错误：

ufuncify_eqs = ufuncify([u,v], numerical_eqs)

我收到的错误消息以：

CodeGenArgumentListError: ('out_817415551552344190', [<sympy.utilities.codegen.OutputArgument object at 0x7f8da54dccd0>])

有人知道ufuncify向量函数的正确方法是什么？

Answer 1

我不认为 ufuncify 支持 SymPy 矩阵作为表达式。您可以单独执行矩阵中的每个表达式。