一、@cache裝飾器的基本概念

@cache 是 Python 3.9 引入的新功能，來(lái)源于標(biāo)準(zhǔn)庫(kù) functools 模塊。它的作用是 對(duì)函數(shù)結(jié)果進(jìn)行緩存（記憶化 memoization），從而避免重復(fù)計(jì)算，提高程序性能。

它的定義方式非常簡(jiǎn)單：

from functools import cache

@cache
def some_function(...):
    ...

當(dāng)你用 @cache 修飾一個(gè)函數(shù)時(shí)，Python 會(huì)將該函數(shù)的輸入?yún)?shù)與計(jì)算結(jié)果自動(dòng)保存到一個(gè)緩存（字典）中。當(dāng)后續(xù)調(diào)用該函數(shù)且參數(shù)完全相同時(shí)，不會(huì)再次執(zhí)行函數(shù)體的計(jì)算邏輯，而是直接返回之前緩存的結(jié)果。

二、工作原理

@cache 裝飾器內(nèi)部其實(shí)是基于 functools.lru_cache 實(shí)現(xiàn)的，它等價(jià)于：

from functools import lru_cache

cache = lru_cache(maxsize=None)

也就是說(shuō)：

• 無(wú)限大小緩存：maxsize=None 表示緩存結(jié)果沒(méi)有數(shù)量限制，會(huì)根據(jù)需要一直增長(zhǎng)（直到你的內(nèi)存允許）。
• 鍵值匹配：緩存根據(jù)參數(shù) 值 判斷是否命中（必須是可哈希類型，如數(shù)字、字符串、元組等）。
• 返回緩存結(jié)果：命中緩存時(shí)，直接返回結(jié)果而不執(zhí)行函數(shù)計(jì)算。

三、使用示例

1. 基本示例：遞歸斐波那契數(shù)列

from functools import cache

@cache
def fib(n):
    print(f"計(jì)算 fib({n})")
    if n < 2:
        return n
    return fib(n - 1) + fib(n - 2)

print(fib(10))  # 第一次需要計(jì)算
print(fib(10))  # 第二次直接命中緩存，不會(huì)重新計(jì)算

運(yùn)行效果：

• 第一次調(diào)用會(huì)完整計(jì)算 fib(10)，輸出多行 "計(jì)算 fib(n)" 的提示信息。
• 第二次調(diào)用相同參數(shù)，直接返回緩存值，不輸出任何計(jì)算日志。

2. 模擬耗時(shí)計(jì)算

import time
from functools import cache

@cache
def slow_function(x):
    print(f"開始計(jì)算 {x}...")
    time.sleep(2)  # 模擬耗時(shí)
    return x * x

# 首次調(diào)用耗時(shí)
print(slow_function(5))

# 第二次調(diào)用相同參數(shù)時(shí)幾乎立即返回
print(slow_function(5))

在實(shí)際場(chǎng)景中，如果一個(gè)函數(shù)計(jì)算代價(jià)高（如數(shù)據(jù)庫(kù)查詢、大文件解析、復(fù)雜計(jì)算等），使用 @cache 能顯著減少重復(fù)工作。

四、注意事項(xiàng)

適用場(chǎng)景

• 適合純函數(shù)（函數(shù)結(jié)果只依賴輸入?yún)?shù)，且不會(huì)產(chǎn)生副作用）。
• 不適合需要每次實(shí)時(shí)更新的函數(shù)（例如獲取當(dāng)前時(shí)間、外部數(shù)據(jù)接口）。

參數(shù)限制

• 參數(shù)必須可哈希（hashable），如整數(shù)、字符串、元組等。
• 如果傳入不可哈希類型（如列表、字典），會(huì)報(bào)錯(cuò) TypeError: unhashable type。

內(nèi)存占用

• @cache 默認(rèn)緩存無(wú)限大小，可能造成內(nèi)存占用增加；如果希望限制緩存大小，應(yīng)使用 lru_cache(maxsize=...) 來(lái)控制。

緩存失效

• @cache 沒(méi)有內(nèi)置超時(shí)或自動(dòng)失效機(jī)制，緩存會(huì)一直保留直到程序結(jié)束或手動(dòng)清除：

your_function.cache_clear()

五、@cache與@lru_cache的區(qū)別

六、總結(jié)

• @cache 能讓函數(shù)的重復(fù)調(diào)用變得快速，尤其適合計(jì)算昂貴且參數(shù)唯一性高的場(chǎng)景。
• 它是 @lru_cache(maxsize=None) 的簡(jiǎn)化版，功能單一但使用更簡(jiǎn)潔。
• 使用時(shí)需注意內(nèi)存占用和參數(shù)類型，確保函數(shù)是“純”的（無(wú)副作用）。
• 對(duì)于需要緩存大小控制的場(chǎng)景或更復(fù)雜的緩存管理，應(yīng)使用 lru_cache。

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評(píng)論