如何创建functools.lru_cache的别名以进行记忆？

Question

要在Python中为函数添加memoization,functools.lru_cache()可以使用装饰器:

import functools

@functools.lru_cache(maxsize=None)
def my_func():
    ...

但是,我想为上面的用法定义以下别名,functools.lru_cache()以使代码更具可读性:

@memoize
def my_func():
    ...

我的问题是:如何定义这样的别名？以下代码不起作用:

memoize = functools.partial(functools.lru_cache, maxsize=None)

Answer 1

你不需要functools.partial它,因为它已经设置为接受两次调用.只需称呼一次:

memoize = functools.lru_cache(maxsize=None)

然后使用别名作为装饰器:

@memoize
def my_func():
    ...