Python时间测量函数

Question

我想创建一个python函数来测试每个函数花费的时间并打印它的名字和它的时间，我如何打印函数名，如果有其他方法请告诉我

def measureTime(a):
    start = time.clock() 
    a()
    elapsed = time.clock()
    elapsed = elapsed - start
    print "Time spent in (function name) is: ", elapsed

Answer 1

首先，我强烈建议使用 profiler或至少使用 timeit .

但是，如果您想编写自己的计时方法来严格学习，这里是开始使用装饰器的地方。

Python 2:

def timing(f):
    def wrap(*args):
        time1 = time.time()
        ret = f(*args)
        time2 = time.time()
        print '%s function took %0.3f ms' % (f.func_name, (time2-time1)*1000.0)
        return ret
    return wrap

而且用法很简单，使用@timing装饰器即可:

@timing
def do_work():
  #code

Python 3:

def timing(f):
    def wrap(*args, **kwargs):
        time1 = time.time()
        ret = f(*args, **kwargs)
        time2 = time.time()
        print('{:s} function took {:.3f} ms'.format(f.__name__, (time2-time1)*1000.0))

        return ret
    return wrap

注意，我调用 f.func_name 来获取函数名作为字符串(在 Python 2 中)，或在 Python 3 中以 f.__name__ 形式获取。