Python3 基础:装饰器和偏函数

来源:互联网 发布:js获取两位小数函数 编辑:程序博客网 时间:2024/06/05 23:41

装饰器

由于函数也是一个对象,而且函数对象可以被赋值给变量,所以,通过变量也能调用该函数。
函数对象有一个__name__属性,可以拿到函数的名字.
在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator)。本质上,decorator就是一个返回函数的高阶函数。
看代码:

def log(fun):def wrapper(*args,**kw):    print("call %s" % fun.__name__)    return fun(*args,**kw)return wrapper@logdef now():    print("2016-12-28")

当我们在调用函数now() 输出结果.

call now2016-12-28

分析一下:
@log就相当于 now = log(now) ,那么此时now是wrapper函数,你在now后面加个()就相当于调用wrapper函数,wrapper函数是调用原本的函数,就是now()所以就会输出了.
由于@log就相当于 now = log(now),当我们调用 log(now) 的时候返回的是wrapper函数,因此此时now.__name__ 就是wrapper.所以,需要把原始函数的__name__等属性复制到wrapper()函数中,否则,有些依赖函数签名的代码执行就会出错。Python内置的functools.wraps就可以做这件事.所以最终的写法:

import functoolsdef log(func):    @functools.wraps(func)    def wrapper(*args, **kw):        print('call %s():' % func.__name__)        return func(*args, **kw)    return wrapper

偏函数

Python的functools模块提供了很多有用的功能,其中一个就是偏函数(Partial function)。要注意,这里的偏函数和数学意义上的偏函数不一样。
假如我们想把字符串按二进制转换成整数,可以这么写:

>>> int('100',base = 2)4>>>

或者你还可以定义函数:

def int2(x,base=2):    return int(x,base)

还有一种更简单的方法,我们可以使用functools.partial帮助我们创建一个偏函数:

>>> import functools>>> int2 = functools.partial(int, base=2)>>> int2('1000000')64

functools.partial的作用就是,把一个函数的某些参数给固定住(也就是设置默认值),返回一个新的函数,调用这个新函数会更简单。

最后,创建偏函数时,实际上可以接收函数对象、*args和**kw这3个参数,当传入:

int2 = functools.partial(int, base=2)

实际上固定了int()函数的关键字参数base,也就是:

int2('10010')

相当于:

kw = { 'base': 2 }int('10010', **kw)

当传入:

max2 = functools.partial(max, 10)

实际上会把10作为*args的一部分自动加到左边,也就是:

max2(5, 6, 7)

相当于:

args = (10, 5, 6, 7)max(*args)

结果为10。

0 0
原创粉丝点击