functools包

2019-11-07  本文已影响0人  Tomtoms

functools

以python3.7为例functools包含如下

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 文档

学习/python

上一篇下一篇

猜你喜欢

热点阅读