不止一个用于lambdify的模块

Question

我试图使lambdify理解使用modules关键字参数期望多种类型的输入.根据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

我在这做错了什么？不应该接受numpy和sympy两种类型吗？任何帮助赞赏!!

Answer 1

该文档表示，该modules参数将为首先出现的模块提供更高的优先级，在本例中为"numpy". 因此，如果两个模块具有相同的功能，则始终采用第一个模块。

一个好的方法是：

import numpy as np

def func(x):
    if isinstance(x, np.ndarray):
        return np.exp(x)
    else:
        return sympy.exp(x)