Я хотел бы ко времени блок кода, не помещая его в отдельную функцию. например:
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 модулем более чистым способом, но я не хочу делать отдельную функцию для блока кода.
спасибо.
Это можно сделать с помощью оператора 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
#...
Вы можете настроить переменную для ссылки на блок кода, который вы хотите отсчитать, поместив этот блок в тройные кавычки Python. Затем используйте эту переменную при создании экземпляра объекта timeit. В некоторой степени следуя примеру из Python timeit docs , я придумал следующее:
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 - текущее время.)