Python 缓存：如何通过有效的缓存来加速代码

php中文网 2024-12-10 15:04:11

此博客最初发布到 crawlbase 博客

高效、快速的代码对于在软件应用程序中创建出色的用户体验非常重要。用户不喜欢等待缓慢的响应，无论是加载网页、训练机器学习模型还是运行脚本。加快代码速度的一种方法是缓存。

缓存的目的是临时缓存经常使用的数据，以便您的程序可以更快地访问它，而不必多次重新计算或检索它。缓存可以加快响应时间、减少负载并改善用户体验。

本博客将介绍缓存原理、其作用、用例、策略以及 python 中缓存的实际示例。让我们开始吧！

在 python 中实现缓存

在 python 中可以通过多种方式进行缓存。我们来看两种常见的方法：使用手动装饰器进行缓存和python内置的functools.lru_cache。

立即学习“Python免费学习笔记（深入）”；

1. 用于缓存的手动装饰器

装饰器是一个包装另一个函数的函数。我们可以创建一个缓存装饰器，将函数调用的结果存储在内存中，并在再次调用相同的输入时返回缓存的结果。这是一个例子：

import requests

# manual caching decorator
def memoize(func):
    cache = {}
    def wrapper(*args):
        if args in cache:
            return cache[args]
        result = func(*args)
        cache[args] = result
        return result
    return wrapper

# function to get data from a url
@memoize
def get_html(url):
    response = requests.get(url)
    return response.text

# example usage
print(get_html('https://crawlbase.com'))

在此示例中，第一次调用 get_html 时，它会从 url 获取数据并缓存。在使用相同 url 的后续调用中，将返回缓存的结果。

使用python的functools.lru_cache

python 在 functools 模块中提供了一个名为 lru_cache 的内置缓存机制。该装饰器会缓存函数调用，并在缓存已满时删除最近最少使用的项目。使用方法如下：

from functools import lru_cache

@lru_cache(maxsize=128)
def expensive_computation(x, y):
    return x * y

# Example usage
print(expensive_computation(5, 6))

在这个例子中，lru_cache缓存了expense_computation的结果。如果使用相同的参数再次调用该函数，它将返回缓存的结果而不是重新计算。