【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来修改时间精度。