我正在尝试使用 modules 关键字参数让 lambdify 理解期望不止一种类型的输入。根据 lambdify ( http://docs.sympy.org/dev/_modules/sympy/utilities/lambdify.html ) 的源代码,这可以通过使用参数列表来完成,但我无法这样做。
import sympy
from sympy import lambdify
x,y=sympy.symbols('x y')
from sympy.parsing.sympy_parser import parse_expr
func=lambdify(x,parse_expr(exp(x)),modules=["numpy","sympy"])
func(array([3,4]))
给予
array([ 20.08553692, 54.59815003])
但是当我尝试
func(y)
我得到一个
Attribute error:exp
我在这里做错了什么? func 不应该同时接受 numpy 和 sympy 类型吗? 任何帮助表示赞赏!!
最佳答案
模块不分派(dispatch)或类似的东西。 lambdify 的工作方式是创建
lambda x: exp(x)
其中 exp
来自您选择的模块的 namespace 。 lambdify(x, exp(x), ['numpy', 'sympy'])
大致等同于
from sympy import *
from numpy import *
# Various name replacements for differences in numpy naming conventions, like
# asin = arcsin
return lambda x: exp(x)
如果你想提供一个自定义的调度函数,你可以使用类似 Saullo Castro 的例子。您还可以通过提供字典将其与 lambdify 一起使用,例如
import numpy as np
import sympy
def myexp(x):
if isinstance(x, np.ndarray):
return np.exp(x)
else:
return sympy.exp(x)
func = lambdify(x, exp(x), [{'exp': myexp}, 'numpy'])
这给出了
>>> func(np.array([1, 2]))
array([ 2.71828183, 7.3890561 ])
>>> func(sympy.Symbol('y'))
exp(y)
关于python - sympy 中有多个用于 lambdify 的模块,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25495375/