86

我想测量在 Python 程序中评估一段代码所用的时间，可能在用户 cpu 时间、系统 cpu 时间和已用时间之间分开。

我知道timeit模块，但是我有很多自己编写的函数，在设置过程中传递它们并不是很容易。

我宁愿有一些可以使用的东西，例如：

#up to here I have done something....
start_counting() #or whatever command used to mark that I want to measure
                   #the time elapsed in the next rows
# code I want to evaluate
user,system,elapsed = stop_counting() #or whatever command says:
                                      #stop the timer and return the times

用户和系统 CPU 时间不是必需的（尽管我想测量它们），但是对于经过的时间，我希望能够做这样的事情，而不是使用复杂的命令或模块。

4

6 回答 6

175

要以秒为单位获取经过的时间，您可以使用timeit.default_timer()：

import timeit
start_time = timeit.default_timer()
# code you want to evaluate
elapsed = timeit.default_timer() - start_time

timeit.default_timer()用于代替time.time()或time.clock()因为它将为任何平台选择具有更高分辨率的计时功能。

于 2013-03-29T16:19:00.707 回答
27

我总是使用装饰器为现有函数做一些额外的工作，包括获取执行时间。它是pythonic和简单的。

import time

def time_usage(func):
    def wrapper(*args, **kwargs):
        beg_ts = time.time()
        retval = func(*args, **kwargs)
        end_ts = time.time()
        print("elapsed time: %f" % (end_ts - beg_ts))
        return retval
    return wrapper

@time_usage
def test():
    for i in xrange(0, 10000):
        pass

if __name__ == "__main__":
    test()
于 2013-03-29T16:36:15.363 回答
11

您可以通过上下文管理器实现此目的，例如：

from contextlib import contextmanager
import time
import logging
@contextmanager
def _log_time_usage(prefix=""):
    '''log the time usage in a code block
    prefix: the prefix text to show
    '''
    start = time.time()
    try:
        yield
    finally:
        end = time.time()
        elapsed_seconds = float("%.2f" % (end - start))
        logging.debug('%s: elapsed seconds: %s', prefix, elapsed_seconds)

使用示例：

with _log_time_usage("sleep 1: "):
    time.sleep(1)
于 2016-05-25T06:57:22.893 回答
11

我发现自己一次又一次地解决了这个问题，所以我最终为它创建了一个库。安装pip install timer_cm。然后：

from time import sleep
from timer_cm import Timer

with Timer('Long task') as timer:
    with timer.child('First step'):
        sleep(1)
    for _ in range(5):
        with timer.child('Baby steps'):
            sleep(.5)

输出：

Long task: 3.520s
  Baby steps: 2.518s (71%)
  First step: 1.001s (28%)
于 2017-04-21T13:21:13.123 回答
1

为了简单起见，我现在非常喜欢另一种选择 - ipython。在 ipython 你有很多有用的东西加上：

%time <expression>- 在表达上获得直接的 cpu 和壁时间

%timeit <expression>- 在表达式循环中获取 cpu 和 wall time

于 2017-11-29T12:27:00.073 回答
0

Python 3 - 使用标准库的简单解决方案

选项 1：三引号代码

import inspect
import timeit


code_block = inspect.cleandoc("""
    base = 123456789
    exponent = 100
    return base ** exponent
    """)
print(f'\Code block: {timeit.timeit(code_block, number=1, globals=globals())} elapsed seconds')

inspect.cleandoc处理多余的制表符和空格的删除，以便可以复制和粘贴代码块而不会出现缩进错误。

 

选项 2：将代码块放在函数中

import timeit


def my_function():
    base = 123456789
    exponent = 100
    return base ** exponent


if __name__ == '__main__':
    print(f'With lambda wrapper: {timeit.timeit(lambda: my_function(), number=1)} elapsed seconds')

请注意，与直接为函数体计时相比，函数调用会增加额外的执行时间。

于 2020-05-03T17:17:23.120 回答