我已经修饰了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向量函数的正确方法是什么?

最佳答案

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

10-08 09:33
查看更多