【Python】使用with上下文实现计时

在Python常常需要记录一段代码运行了多长时间,普通的一般用两个time.time(),然后计算差值;高级一点的,就是写一个计时装饰器,用装饰器包裹方法,记录运行时间,但这这意味着你的代码需要写在函数里,然后才能使用此计时装饰器。

今天我要介绍的是一种可以给一段代码计时,而不是一个函数计时的方便的、“高级的”的Python计时方法。

代码如下:

import time
from contextlib import contextmanager

@contextmanager
def timer(name):
    start = time.time()
    yield
    print(f'[{name}] done in {time.time() - start:.2f} s')

使用方法:

with timer('Test'):
    i = 0
    while i < 1000000:
        i += 1

输出:

[Test] done in 0.11 s

从1加到1百万需要0.11s。可以在timer通过修改.2f来修改时间精度。