python - 如何找出一个 lambda 函数需要多少个参数

Question

我正在尝试创建一个函数，该函数将根据传入的 lambda 函数进行最小二乘拟合。我想创建一个长度为零的数组，该数组的长度等于 lambda 函数为初始猜测 lambda 函数所采用的参数数量。所以如果它是线性的，我想要[0,0]，而对于二次我想要[0,0,0]。

#polynomial functions
linear = lambda p, x: p[0] * x + p[1]
quadratic = lambda p, x: p[0] * x**2 + p[1] * x + p[2]
cubic = lambda p, x: p[0] * x**3 + p[1] * x**2 + p[2] * x + p[3]

#polynomial functions forced through 0
linear_zero = lambda p, x: p[0] * x 
quadratic_zero = lambda p, x: p[0] * x**2 + p[1] * x 
cubic_zero = lambda p, x: p[0] * x**3 + p[1] * x**2 + p[2] * x

def linFit(x, y,fitfunc):
    errfunc = lambda p, x, y: fitfunc(p, x) - y     

在这里，我想创建一个零数组。但此时 p 尚未定义。所以 len(p) 不起作用。

    init_p = np.array(zeros(len(p)))  #bundle initial values in initial parameters

    p1, success = optimize.leastsq(errfunc, init_p.copy(), args = (x, y))
    return p1   

Answer 1

在 python >= 2.7 下：

>>> l = lambda a, b: None
>>> l.func_code.co_argcount
2

或低于 2.6：

>>> l.__code__.co_argcount
2

Answer 2

通过查看它的代码对象__code__：

>>> p=lambda x,y:x+y
>>> len(p.__code__.co_varnames)
2
>>> p.__code__.co_varnames
('x', 'y')