functools包
2019-11-07 本文已影响0人
Tomtoms
functools
以python3.7为例functools包含如下
- cmp_to_key
- get_cache_token
- lru_cache
- namedtuple
- partial
- partialmethod
- recursive_repr
- reduce
- singledispatch
- total_ordering
- update_wrapper
- wraps
partial
Partial通过包装的手法,允许我们重新定义函数名
如果一个函数的参数比较复杂,其中某些参数在项目中已经固定,那么可通过partial重新命名
from functools import *
def add(a, b):
return a + b
add3 = partial(add, 3)
add5 = partial(add, 5)
print(add3(1))
print(add5(1))
total_ordering
针对已经实现了__lt__
,__gt__
,__ge__
这些方法中至少一个,使用该装饰器会自动补全其他,例如
from functools import *
@total_ordering
class square:
def __init__(self, x):
self.x = x
def __lt__(self, other):
return self.x**2 < other.x**2
s = square(2)
s1 = square(3)
print(s > s1)
singledispatch
首先,Python是不支持函数重载的,如果有一个需要进行重载的函数,就需要在一个函数中指派各种类型,意味着需要写大量的if/else语句,singledispatch装饰器可以减轻这种烦恼,比如要写一个将数据/2的函数,int/float/double类型都是/2,str类型是截断一半
from functools import *
import numbers
@singledispatch
def half(obj):
print(obj, type(obj), 'obj')
return obj[:len(obj)//2]
@half.register(str)
def _(text):
print(text, type(text), 'str')
return text[:len(text)//2]
@half.register(int)
@half.register(float)
def _(num):
print(num, type(num), 'number')
return num/2
print(half(12))
print(half('abcdef'))
print(half([1,2,3,4,5,6]))
结果如下
12 <class 'int'> number
6.0
abcdef <class 'str'> str
abc
[1, 2, 3, 4, 5, 6] <class 'list'> obj
[1, 2, 3]
lru_cache
一个缓存装饰器,用于缓存相同调用的结果,由于用了字典存储缓存,该函数的固定参数
和关键字
必须是可哈希的。f(a=1, b=2)和f(b=2, a=1)会被缓存两次,f(a=1)和f(a=1.0)也会被缓存两次。
import time
import urllib.request
from functools import *
def timethis(func):
'''
Decorator that reports the execution time.
'''
@wraps(func)
def wrapper(*args, **kwargs):
start = time.time()
result = func(*args, **kwargs)
end = time.time()
print(func.__name__, end-start)
return result
return wrapper
@timethis
@lru_cache(maxsize=None)
def get(url):
return urllib.request.urlopen(url).read()
get('http://www.baidu.com')
get('http://www.baidu.com')
get('http://www.baidu.com')
get('http://www.baidu.com')
get('http://www.baidu.com')
结果如下
get 0.056482791900634766
get 1.9073486328125e-06
get 9.5367431640625e-07
get 0.0
get 0.0
recursive_repr
wraps
cmp_to_key
python3 为什么取消了sort方法中的cmp参数? - 知乎
get_cache_token
namedtuple
collections — 容器数据类型 — Python 3.7.3 文档