如何將functools.lru_cache應用於具有可變參數的函數？

Question

我有一個參數爲numpy.ndarray的函數。它是可變的，所以它不能被lru_cache緩存。

這是否有任何現有的解決方案？

Answer 1

可能這樣做的最簡單方法是記憶只有不可變對象的版本。

說你的函數需要一個np.array，我們假設它是一個1d數組。幸運的是，它很容易轉換爲tuple：

import numpy as np 

a = np.array([1, 2, 3, 4]) 
>> tuple(a) 
(1, 2, 3, 4) 

，反之亦然：

>> np.array(tuple(a)) 
array([1, 2, 3, 4]) 

所以你喜歡的東西

# Function called by the rest of your program 
array_foo(a) # `a` is an `np.array` 
    ... 
    return tuple_foo(tuple(a)) 

然後memoize的不是此函數：

# Internal implementation 
@functools.lru_cache 
tuple_foo(t) # `t` is a tuple 
    ... 
    a = np.array(t)