使用 lru_cache 装饰器可缓存函数结果,避免重复计算,提升性能;通过 maxsize 控制缓存大小,适用于纯函数如递归、数学运算;也可用字典或类封装实现自定义缓存,便于灵活管理。

在 Python 中,缓存函数结果是一种有效减少重复计算的手段,特别适用于那些输入不变时输出也不变、但计算成本较高的函数。通过记忆已计算的结果,可以在后续相同输入时直接返回结果,避免重复执行。
使用 functools.lru_cache 装饰器
lru_cache 是 Python 标准库中 functools 模块提供的一个装饰器,能自动缓存函数的返回值。它使用 LRU(最近最少使用)算法管理缓存大小,防止内存无限增长。
用法示例:
from functools import lru_cache@lru_cache(maxsize=128)def fibonacci(n):if n <= 1:return nreturn fibonacci(n - 1) + fibonacci(n - 2)
print(fibonacci(100)) # 第一次计算并缓存print(fibonacci(100)) # 直接从缓存读取
说明:
立即学习“Python免费学习笔记(深入)”;
添加 @lru_cache 装饰器即可启用缓存maxsize 控制缓存条目数,设为 None 表示无限制适合递归函数、数学计算等场景
自定义缓存字典实现
对于需要更灵活控制的场景,可以用字典手动实现缓存逻辑。
cache = {}def expensive_function(x, y):key = (x, y)if key in cache:return cache[key]
result = sum(i * j for i in range(x) for j in range(y)) # 模拟耗时计算cache[key] = resultreturn result
这种方式的优点是你可以完全控制缓存的生成、清除和存储结构,比如按参数类型区分缓存,或加入过期机制。
使用类封装带缓存的方法
当多个函数或状态相关时,可将缓存逻辑封装在类中。
class Calculator: def __init__(self): self.cache = {}def compute(self, data): key = tuple(data) if key not in self.cache: self.cache[key] = sum(x ** 2 for x in data) return self.cache[key]
这种方法便于管理实例级别的缓存,也支持调用 clear() 方法手动清理缓存。
注意事项与适用场景
缓存虽好,但也需注意以下几点:
函数必须是纯函数:相同输入永远返回相同输出不可缓存涉及随机性、时间、外部状态变化的函数缓存会占用内存,设置合理的 maxsize 很重要不可哈希的参数(如列表、字典)不能用于 lru_cache,需转换为元组或冻结结构
基本上就这些。合理使用缓存能显著提升程序性能,尤其是在处理递归、动态规划、频繁调用的工具函数时效果明显。
以上就是Python 缓存函数结果减少重复计算的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1372713.html
微信扫一扫
支付宝扫一扫