解压 sympylambdaify 函数

Question

下面的代码我有。

from sympy import * 
  
x = symbols('x')
expr = sin(x)
     
# Use sympy.lambdify() method
f = lambdify(x, expr, "math") 

如果 print(f) 给出 ''，有没有办法取回表达式(sin(x)) 来自 f?

Answer 1

help(f) 显示：

Help on function _lambdifygenerated:

_lambdifygenerated(x)
    Created with lambdify. Signature:
    
    func(x)
    
    Expression:
    
    sin(x)
    
    Source code:
    
    def _lambdifygenerated(x):
        return sin(x)

f.__doc__ 是相同的字符串。

由于我们指定了“math”，标量可以工作，但数组不行：

In [12]: f(1.23)
Out[12]: 0.9424888019316975

In [13]: f(np.arange(3))
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Input In [13], in <module>
----> 1 f(np.arange(3))

File <lambdifygenerated-2>:2, in _lambdifygenerated(x)
      1 def _lambdifygenerated(x):
----> 2     return sin(x)

TypeError: only size-1 arrays can be converted to Python scalars

Answer 2

我进入了 Sympy 的源代码，发现了这一行，其中羔羊化函数的内容在变量 funcstr 中定义：

https://github.com/sympy/sympy/blob/88b9cba9d7e633ec769baf1fc5267acfd7f54788/sympy/utilities/lambdify.py#L863

# Create the function definition code and execute it
funcname = '_lambdifygenerated'
if _module_present('tensorflow', namespaces):
    funcprinter = _TensorflowEvaluatorPrinter(printer, dummify) # type: _EvaluatorPrinter
else:
    funcprinter = _EvaluatorPrinter(printer, dummify)

if cse == True:
    from sympy.simplify.cse_main import cse as _cse
    cses, _expr = _cse(expr, list=False)
elif callable(cse):
    cses, _expr = cse(expr)
else:
    cses, _expr = (), expr
funcstr = funcprinter.doprint(funcname, iterable_args, _expr, cses=cses)

目前 Sympy 不返回 funcstr，但我在本地安装中修改了它，这样它就可以了，而且似乎成为你所追求的。

import sympy as sy 
  
x = sy.symbols('x')
expr = sy.sin(x)
     
# Use sympy.lambdify() method
f, funcstr = sy.lambdify(x, expr, "math") 

funcstr.split(":")[-1].split("return")[-1].strip()
>>> 'sin(x)'

我获取内容的方式有点狡猾，而且不一定可靠，但我希望它能为您提供一个工作起点。

请注意，如果您使用参数 cse，则此提取将不准确，因为它将错过之前的简化。