Python + inspect 一行实现递归 fib 函数-工具盒子

背景 {#背景}

有个裙友要看看用 lambda 能不能在一行里定义出来 fib 函数，并且不要那个根号五的数学公式，于是就有了这篇文章。

介绍 {#介绍}

inspect 库可以帮助我们拿到 Python 上下文的各种信息，自然也包括了当前正在运行的函数。配合 eval 可以达到我们的目的。

实现 {#实现}

原始方法 {#原始方法}

虽然大家都知道，但还是放上来做一个对比。

|-----------------------|--------------------------------------------------------------------------------------------| | 1 2 3 4 5 6 7 | def fib(n): if n in (1, 2): return 1 return fib(n-1) + fib(n-2) ## 8 print(fib(6)) |

通过 inspect 获取当前函数 {#通过-inspect-获取当前函数}

我目前只知道 codeobject 配合 eval 来执行的方法，如下：

|---------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 5 6 7 8 9 | import inspect def fib(): if n in (1, 2): return 1 return eval(inspect.currentframe().f_code, {'n': n-1, 'inspect': inspect}) + \ eval(inspect.currentframe().f_code, {'n': n-2, 'inspect': inspect}) ## 8 print(eval(fib.__code__, {'n': 6, 'inspect': inspect})) |

注意，此时我们就通过给 eval 的第二个参数指定函数运行时的环境（即globals），这样我们就可以在函数内部直接通过变量名访问了。

用 lambda 写成一行 {#用-lambda-写成一行}

还用了 __import__ 来导包，是真正的一行捏：

|------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 5 6 7 8 9 10 11 12 | result = (lambda number: eval( (lambda: 1 if n in (1, 2) else eval(f().f_code, {'n': n-1, 'f': f}) + eval(f().f_code, {'n': n-2, 'f': f}) ).__code__, {'n': number, 'f': __import__('inspect').currentframe} ) )(6) ## 8 print(result) |

注意不要改我们 lambda number 后边的参数名，如果也改成 n 的话，eval 会报错，下面是报错信息：

|---------------------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 5 6 | Traceback (most recent call last): File "test.py", line 1, in <module> result = ( File "test.py", line 3, in <lambda> eval((lambda: 1 if n in (1, 2) TypeError: code object passed to eval() may not contain free variables |

因为此时内层函数用的 n 就类似 Lua 的 upvalue，或者 nonlocal，总之就是闭包里面的那个概念，它无法访问到就报错了。