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