打印定义lambda函数的代码

时间:2008-12-02 17:57:23

标签: python

我希望能够打印lambda函数的定义代码。

如果我通过lambda语法定义此函数的示例:

>>>myfunction = lambda x: x==2
>>>print_code(myfunction)

我想得到这个输出:

x==2

6 个答案:

答案 0 :(得分:60)

只要将代码保存到源文件,就可以检索到源代码 使用检查模块的对象。

例如: 打开编辑器类型:

myfunction = lambda x: x==2

另存为lamtest.py

打开shell类型python来获取交互式python 输入以下内容:

>>>from lamtest import myfunc
>>>import inspect
>>>inspect.getsource(myfunc)

结果:

'myfunc = lambda x: x==2\n'

答案 1 :(得分:21)

它仅适用于基于数学的操作,但您可能会查看SymPyLambda()对象。它的设计正是为了这个目的:

>>> from sympy import *
>>> x = Symbol('x')
>>> l = Lambda(x, x**2)
>>> l
Lambda(_x, _x**2)
>>> l(3)
9

它甚至支持漂亮的打印:

>>> pprint(l)
 ⎛    2⎞
Λ⎝x, x ⎠

要做你的等于示例,请使用SymPy Eq()对象:

>>> l1 = Lambda(x, Eq(x, 2))
>>> l1
Lambda(_x, _x == 2)
>>> l1(2)
True

它支持部分参数扩展:

>>> y = Symbol('y')
>>> l2 = Lambda((x, y), x*y + x)
>>> l2(1)
Lambda(_y, 1 + _y)
>>> l2(1, 2)
3

当然,你可以获得所有SymPy计算机代数的优势:

>>> l3 = Lambda(x, sin(x*pi/3))
>>> pprint(l3(1))
  ⎽⎽⎽
╲╱ 3 
─────
  2  

顺便说一下,如果这听起来像一个无耻的插头,那是因为它是。我是SymPy的开发人员之一。

答案 2 :(得分:13)

虽然我普遍认为inspect是一个很好的答案,但我不同意你无法获得解释器中定义的对象的源代码。如果您使用dill中的dill.source.getsource,则可以获得函数和lambdas的来源,即使它们是以交互方式定义的。 它还可以从curries中定义的绑定或非绑定类方法和函数中获取代码...但是,如果没有封闭对象的代码,您可能无法编译该代码。

>>> from dill.source import getsource
>>> 
>>> def add(x,y):
...   return x+y
... 
>>> squared = lambda x:x**2
>>> 
>>> print getsource(add)
def add(x,y):
  return x+y

>>> print getsource(squared)
squared = lambda x:x**2

>>> 
>>> class Foo(object):
...   def bar(self, x):
...     return x*x+x
... 
>>> f = Foo()
>>> 
>>> print getsource(f.bar)
def bar(self, x):
    return x*x+x

>>> 

答案 3 :(得分:5)

这将非常困难,因为你的lambda函数将被编译为字节码,而你的myfunction对象只会指向字节码,而不是你编写的人类可读代码。

例如,如果您定义了2个函数,一个使用lambda语法,另一个使用def语句,如下所示:

>>> lambda_func = lambda x: x==2
>>> def def_func(x): return x == 2
...

就python而言,这两个对象(lambda_func和def_func)将是等价的。事实上,如果你继续使用dis module(如同建议的rebra)对它们进行反汇编,你会得到相同的结果:

>>> import dis
>>> dis.dis(lambda_func)
  1           0 LOAD_FAST                0 (x)
              3 LOAD_CONST               1 (2)
              6 COMPARE_OP               2 (==)
              9 RETURN_VALUE
>>> dis.dis(def_func)
  1           0 LOAD_FAST                0 (x)
              3 LOAD_CONST               1 (2)
              6 COMPARE_OP               2 (==)
              9 RETURN_VALUE

在这种情况下,您可以看到当它是多对一关系时如何获取原始代码

答案 4 :(得分:2)

这是怎么回事?

class MyLambda( object ):
    def __init__( self, body ):
        self.body= body
    def __call__( self, arg ):
        x = arg
        return eval( self.body )
    def __str__( self ):
        return self.body

f= MyLambda( "x == 2" )
print f(1)
print f(2)
print f

答案 5 :(得分:-4)

你为什么要这样做?

我猜你可以使用“dis”模块将你的代码反汇编为python字节码,但它可能不是你想要的。

http://www.python.org/doc/2.5.2/lib/module-dis.html

同样,我看不到这个用例。也许eval()更适合您的问题。它是内置的,然后您可以使用字符串在代码中传递代码。