python - Sympy - ufuncify 向量函数

Question

我有 lambdify-ied 一个修改后的向量函数 FitzHugh–Nagumo model (没有扩散项):

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 会导致错误:

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

我收到的错误消息以:

结尾

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

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

Answer 1

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