Python 中的计时代码块，无需将其放入函数中

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

    #...

Answer 2

您可以通过将代码块放在 Python 的三引号内来设置一个变量来引用您想要计时的代码块。然后在实例化 timeit 对象时使用该变量。有点遵循 Python timeit 文档 中的示例，我想出了以下内容：

import timeit
code_block = """\
total = 0
for cnt in range(0, 1000):
    total += cnt
print total
"""
tmr = timeit.Timer(stmt=code_block)
print tmr.timeit(number=1)

为我打印：

499500

0.000341892242432

（其中499500是定时块的输出，0.000341892242432是运行的时间。）

Answer 3

根据 Skilldrick answer 有一个愚蠢的模块，我认为可以提供帮助：BlockLogginInator。

import time
from blocklogginginator import logblock  # logblock is the alias

with logblock(name='one second'):
    """"
    Our code block.
    """
    time.sleep(1)

---

>>> Block "one second" started 
>>> Block "one second" ended (1.001)