Python将lambda参数指定为函数参数

时间:2016-03-23 14:45:24

标签: python lambda

是否有可能指定lambda作为函数参数可以使用多少个参数?

例如:

def func(k=lambda x:return x**2):
   return k

如果k不是标准的lambda,我可以指定k应该只接受一个参数吗?

最高

1 个答案:

答案 0 :(得分:6)

你可以使用inspect.getargspec执行类似的操作:

import inspect
def func (k = lambda x: x ** 2):
    if not callable(k) or len(inspect.getargspec(k).args) != 1:
        raise TypeError('k must take exactly one argument.')
    # Do whatever you want

请注意,上面的内容很可能会失败(但不应该):

func (lambda x, y = 8: x + y)

...如果你想处理这种情况,你需要更复杂的东西:

import inspect
def func (k = lambda x: x ** 2):
    if not callable(k):
        raise TypeError('k must be callable.')
    argspec = inspect.getfullargspec(k)
    nargs = len(argspec.args)
    ndeft = 0 if argspec.defaults is None else len(argspec.defaults)
    if nargs != ndeft + 1:
        raise TypeError('k must be callable with one argument.')
    # Do whatever you want