为什么添加我自己的函数时我的lambdify不起作用

Question

我创建了一个函数并使其在我的 Sympy 表达式中可用，如下所示：

  def Unit(x):
        if(x != 0):
            return 0
        else:
            return 1
  Unit = Function('Unit')
  x = Symbol('x')

我的表情：

fx = x ** 2 + Unit(x)

但是当我跑步时：

lam_f = lambdify(x, fx, modules=["sympy"])
print(lam_f(-1))

它说我的单位没有定义？ 谁能解释我哪里出错了？

Answer 1

Function('Unit') 返回一个名为 Unit 的未定义函数。见this question。如果你想使用你之前定义的函数Unit，去掉对Function()的调用：

def Unit(x):
    if(x != 0):
        return 0
    else:
        return 1

x = Symbol('x')
fx = x**2 + Unit(x)

lam_f = lambdify(x, fx, modules=['sympy'])
print(lam_f(-1)) # prints 1