python - Python中的代码计时 block 而不将其放入函数中

Question

我想对一段代码进行计时，而不是将它放在一个单独的函数中。例如:

def myfunc:
  # some code here
  t1 = time.time()
  # block of code to time here
  t2 = time.time()
  print "Code took %s seconds." %(str(t2-t1))

但是，我想以更简洁的方式使用 timeit 模块来执行此操作，但我不想为代码块创建一个单独的函数。

谢谢。

Answer 1

您可以使用 with 语句来完成此操作。例如:

import time    
from contextlib import contextmanager

@contextmanager  
def measureTime(title):
    t1 = time.clock()
    yield
    t2 = time.clock()
    print '%s: %0.2f seconds elapsed' % (title, t2-t1)

像这样使用:

def myFunc():
    #...

    with measureTime('myFunc'):
        #block of code to time here

    #...