python自定义函数详解_Python自定义函数

自定义函数

使用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