Python的functools模組使用及說明

2022-07-14 14:02:23

partial

用於建立一個偏函數，將預設引數包裝一個可呼叫物件，返回結果也是可呼叫物件。

偏函數可以固定住原函數的部分引數，從而在呼叫時更簡單。

from functools import partial

int2 = partial(int, base=8)
print(int2('123'))
# 83

update_wrapper

使用 partial 包裝的函數是沒有__name__和__doc__屬性的。

update_wrapper 作用：將被包裝函數的__name__等屬性，拷貝到新的函數中去。

from functools import update_wrapper
def wrap2(func):
    def inner(*args):
        return func(*args)
    return update_wrapper(inner, func)

@wrap2
def demo():
    print('hello world')

print(demo.__name__)
# demo

wraps

warps 函數是為了在裝飾器拷貝被裝飾函數的__name__。

就是在update_wrapper上進行一個包裝

from functools import wraps
def wrap1(func):
    @wraps(func)    # 去掉就會返回inner
    def inner(*args):
        print(func.__name__)
        return func(*args)
    return inner

@wrap1
def demo():
    print('hello world')

print(demo.__name__)
# demo

reduce

在 Python2 中等同於內建函數 reduce

函數的作用是將一個序列歸納為一個輸出

reduce(function, sequence, startValue)

from functools import reduce

l = range(1,50)
print(reduce(lambda x,y:x+y, l))
# 1225

cmp_to_key

在 list.sort 和內建函數 sorted 中都有一個 key 引數

x = ['hello','worl','ni']
x.sort(key=len)
print(x)
# ['ni', 'worl', 'hello']

Python3 之前還提供了cmp引數來比較兩個元素

cmp_to_key 函數就是用來將老式的比較函數轉化為 key 函數

lru_cache

允許我們將一個函數的返回值快速地快取或取消快取。

該裝飾器用於快取函數的呼叫結果，對於需要多次呼叫的函數，而且每次呼叫引數都相同，則可以用該裝飾器快取呼叫結果，從而加快程式執行。

該裝飾器會將不同的呼叫結果快取在記憶體中，因此需要注意記憶體佔用問題。

from functools import lru_cache
@lru_cache(maxsize=30)  # maxsize引數告訴lru_cache快取最近多少個返回值
def fib(n):
    if n < 2:
        return n
    return fib(n-1) + fib(n-2)
print([fib(n) for n in range(10)])
fib.cache_clear()   # 清空快取

singledispatch

單分發器， Python3.4新增，用於實現泛型函數。

根據單一引數的型別來判斷呼叫哪個函數。

from functools import singledispatch
@singledispatch
def fun(text):
    print('String：' + text)

@fun.register(int)
def _(text):
    print(text)

@fun.register(list)
def _(text):
    for k, v in enumerate(text):
        print(k, v)

@fun.register(float)
@fun.register(tuple)
def _(text):
    print('float, tuple')
fun('i am is hubo')
fun(123)
fun(['a','b','c'])
fun(1.23)
print(fun.registry)    # 所有的泛型函數
print(fun.registry[int])    # 獲取int的泛型函數
# String：i am is hubo
# 123
# 0 a
# 1 b
# 2 c
# float, tuple
# {<class 'object'>: <function fun at 0x106d10f28>, <class 'int'>: <function _ at 0x106f0b9d8>, <class 'list'>: <function _ at 0x106f0ba60>, <class 'tuple'>: <function _ at 0x106f0bb70>, <class 'float'>: <function _ at 0x106f0bb70>}
# <function _ at 0x106f0b9d8>

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。