自定义函数
使用def定义函数,例如:
def my_abs(x) :
if x >= 0 :
return x
else :
return -x
函数可以返回多个值,但是这是假象,实际上函数返回的是tuple类型的值,只不过返回的tuple值可以省略括号:
def swap(x, y) :
temp = x
x = y
y = temp
return x, y
x, y = swap(x, y)
print(x, y)
函数定义--定义默认参数
def power(x, n=2) : # 默认参数必须在参数的最后
....
函数定义--定义可变参数
def fn(*args) : # 通过添加星号'*',args可以传入任意个参数,实际上args被当做tuple处理
....
装饰器函数
装饰器函数可以为输入函数统一添加新的功能,例如装饰器log可以为制定的函数添加输出log的功能:
def log(f) :
def fn(*args, **kw) :
print( 'call' + f.__name__ + '()...')
return f(*args, **kw)
return fn
# 通过装饰器修正f函数
def f(x, y)
return x + y
f = log(f) # log修正f后,再调用f
print f(x, y)
# 也可以通过@简化装饰器的调用
@log
def f(x, y)
return x + y
print f(x, y) # 不需要再显式地使用log修正f
装饰器函数也可以输入除函数以外的参数,此时装饰器函数的返回值是另一个装饰器函数:
def log(string)
def log_new(f)
def fn(*arg, **kw)
print( '[' + string + ']' + 'call ' + f.__name__ + '()...' )
return f(*arg, **kw)
return fn
return log_new
@log('DEBUG')
def f(x, y)
return x + y
装饰器修正的函数在被调用时,实际上并不是原始函数被调用,而是装饰器中定义的函数被调用,为了使得装饰器中定义的函数的属性和原始函数保持一致,使用python内置函数functools完成函数属性的复制:
def log(f) :
@functools.wraps(f) # 将函数f的属性复制给函数fn
def fn(*args, **kw) :
print( 'call' + f.__name__ + '()...')
return f(*args, **kw)
return fn