这个问题的来源是在看python中的装饰器的时候,例子里给了这样一段代码:

#-*- coding: UTF-8 -*- 
import time    
def foo():
    print 'in foo()'  
# 定义一个计时器,传入一个,并返回另一个附加了计时功能的方法 

def timeit(func):
    # 定义一个内嵌的包装函数,给传入的函数加上计时功能的包装
    def wrapper():
        start = time.clock()
        func()
        end =time.clock()
        print 'used:', end - start

    # 将包装后的函数返回
    return wrapper   
foo = timeit(foo) 
foo() 

上述代码实现了一个装饰器,但没有处理foo()函数带参数以及返回值的情况。为了改进这段代码,首先要理解的是python中参数传递*和**的问题

如果在函数定义时候,按照如下定义的:

#代码段1
def foo(*a):
    pass

或者

#代码段2
def foo(**a):
    pass

这种定义方法相当与C语言中的不定参数,在定义的时候,函数将被传入的参数是未知的,运行的时候,代码段1 的代码 参数将会一tuple的形式组织起来,传入函数; 代码段2 的代码,参数将会以dictionary的形式传入

*和**除了在定义函数的时候有用,它的另一个功能就是对tuple和dict进行展开。

于是我们便可以利用这个运算符对最开始的代码进行改进:


#-*- coding: UTF-8 -*- import time def foo(a): print ' in foo() %s'%a return 100 def foo2(): print ' in foo2()' return #定义定时器,传入一个,并返回另一个附加了计时功能的方法 def timeit(func): #定义一个内嵌的包装函数,给传入的函数加上计时的包装 def wrapper(*args,**args2): ######1######### start = time.clock() res=func(*args,**args2) #########2########### end = time.clock() print 'used:',end - start return res return wrapper a=1 foo = timeit(foo) print foo(a) foo2 = timeit(foo2) foo2()

相比于原代码,主要更改是标号的两行。 第一行在函数定义的时候,参数里添加了一个不定参数,以便接受原函数的参数,第二行利用了*运算符的第二个功能,将参数展开,传递给原函数。


GHOST_349178
42 声望3 粉丝