python 同情 : lambdify returns the wrong answer when function defined piecewise

Question

我对为什么以下代码的答案是 10 而不是 1 感到困惑。有人可以帮助我了解 lambdify 发生了什么或者是什么导致了错误答案吗？

import sympy
from sympy.utilities.lambdify import lambdify
from sympy import Function
from sympy.abc import x, y

def kFct(xIndex,loc,k1,k2):
...   if xIndex <= loc:
...     return k1
...   else:
...     return k2
... 
loc = 0.5
k1 = 1
k2 = 10

kfun = lambdify( (x,y), kFct(x,loc,k1,k2) )
print kfun(0,0)
>>> 10

为什么答案不是 k1 或 1，因为 x = 0 小于 loc = 0.5？

但是，如果我这样做，它会返回正确的答案

print kfct(0,loc,k1,k2)
>>> 1

我需要将 kfun 作为 x 和 y 的函数，因为稍后，我将它用作积分参数的一部分。它最终也将取决于 y。

我在 Mac 10.6.x 上使用 python 2.6.8。

Answer 1

lambdify 的参数在传递之前进行评估，因此您实际上并没有将函数传递给 lambdify，您传递的是数字 10:

>>> kFct(x, loc, k1, k2)
10

你在这里得到 10，因为

>>> x <= loc
x <= 0.5
>>> bool(x <= loc)
False

因此采用第二个分支。由于 Python 的工作方式，我不认为你能够让它工作——你不能只抑制一个分支被采用。 (原则上，一个程序可以做一些疯狂的字节码内省(introspection)，但我很确定 sympy 不会。)

不过，您可以使用 implemented_function:

>>> f = implemented_function(Function('kFct'), lambda x,y: kFct(x, loc, k1, k2))
>>> kfun = lambdify((x,y), f(x,y))
>>> kfun(0,0)
1
>>> kfun(0.5,0)
1
>>> kfun(0.51,0)
10
>>> kfun(1, 0.0)
10

不过，考虑到额外的间接性，我不确定这对您有多大好处:我可能只使用函数本身(假设您最终要寻找积分的数值计算。)