如何在Python中自动测量函数的执行时间

Question

我需要有一个基类 ，我将用它来继承其他类，我想测量其函数的执行时间。

因此， 这一翻译有像这样的 ：

class Worker():
    def doSomething(self):
        start = time.time()
        ... do something
        elapsed = (time.time() - start)
        print "doSomething() took ", elapsed, " time to finish"

#outputs: doSomething() took XX time to finish

我想要这样的东西：

class Worker(BaseClass):
    def doSomething(self):
        ... do something

#outputs the same: doSomething() took XX time to finish

所以BaseClass需要处理测量时间

Answer 1

一种方法是使用装饰器（装饰器的PEP） （关于装饰器的一系列教程文章中的第一篇） 。 这是一个做你想要的例子。

from functools import wraps
from time import time

def timed(f):
  @wraps(f)
  def wrapper(*args, **kwds):
    start = time()
    result = f(*args, **kwds)
    elapsed = time() - start
    print "%s took %d time to finish" % (f.__name__, elapsed)
    return result
  return wrapper

这是其使用的一个例子

@timed
def somefunction(countto):
  for i in xrange(countto):
    pass
  return "Done"

为了说明它是如何工作的，我从python提示符调用了该函数：

>>> timedec.somefunction(10000000)
somefunction took 0 time to finish
'Done'
>>> timedec.somefunction(100000000)
somefunction took 2 time to finish
'Done'
>>> timedec.somefunction(1000000000)
somefunction took 22 time to finish
'Done'

Answer 2

你检查了“配置文件”模块吗？

即，您确定需要实现自己的自定义框架，而不是使用该语言的默认分析机制吗？

您也可以谷歌为“python hotshot”获得类似的解决方案。

Answer 3

还有timeit ，它是标准库的一部分，并且非常易于使用。 记住：不要重新发明轮子！