Python装饰器之lru_cache

在流畅的Python里看到的，做一下笔记。

在运行一些比较耗时的函数，如递归时重复计算相同参数的函数时。通过python自带的装饰器functools.lru_cache可以将函数结果保存起来，作为缓存在重复计算时直接返回结果。减少函数运行的时间。

以斐波那契数列的递归为例，首先实现一个递归的斐波那契方法：

def fib(n):
    if n < 2:
        return n
    return fib(n-1) + fib(n-2)

递归的时间复杂度为O（2^n）,当传入的参数较大时，运行时间会显著增加。
而我们只需要给这个函数加上一个装饰器，一切都变得快了起来。

@functools.lru_cache()
def fib(n):
    if n < 2:
        return n
    return fib(n-1) + fib(n-2)

以传入参数30为例，当不使用装饰器时，运行时间为24秒

加上装饰器后

此外，装饰器支持2个参数，maxsize表示存储多少个结果，typed表示是否将结果根据类型分开保存。
而lru_cache通过字典保存结果，键值根据传入的参数创建，所以装饰的方法传入的参数必须都是可散列的。