Как вычислить время выполнения программы python

Я хочу написать функцию, которая ведет себя точно так же, как данная функция, за исключением того, что она печатает время, затраченное на ее выполнение. Именно так:

>>> fib = profile(fib) >>> fib(20) time taken: 0.1 sec 10946

Это мой код, и он будет печатать сообщение в каждом вызове функции.

Мой код выше будет печатать сообщение “взятое время:…” для каждого фиб (n-1). Таким образом, будет много времени для приема сообщений:… “. Могу ли я найти способ просто распечатать время выполнения fib (20), а не каждое время выполнения fib (n-1)?

Лучший ответ:

Я предполагаю, что ваш вопрос: “Как написать profile чтобы он печатал только одно сообщение, даже когда я украшаю функцию, которая может называть себя?”

Вы можете отслеживать, какие функции в настоящее время синхронизируются. Таким образом, если функция вызывает себя, вы знаете, что вы уже синхронизировали ее по стеку и не нужно ничего делать для второго экземпляра.

def profile(f, currently_evaluating=set()): #’currently_evaluating’ will persist across all decorated functions, due to «mutable default argument» behavior. #see also http://stackoverflow.com/questions/1132941/least-astonishment-in-python-the-mutable-default-argument def g(x): #don’t bother timing it if we’re already doing so if f in currently_evaluating: return f(x) else: start_time = time.clock() currently_evaluating.add(f) try: value = f(x) finally: currently_evaluating.remove(f) end_time = time.clock() print(‘time taken: ‘.format(time=end_time-start_time)) return value return g

Если вы используете 3.X, вы можете отключить изменчивую аргументацию по умолчанию, используя ключевое слово nonlocal .

Ты неправильно замеряешь время в Python! Или нет?

def profile(f): is_evaluating = False def g(x): nonlocal is_evaluating if is_evaluating: return f(x) else: start_time = time.clock() is_evaluating = True try: value = f(x) finally: is_evaluating = False end_time = time.clock() print(‘time taken: ‘.format(time=end_time-start_time)) return value return g