python - timeit

timeit 模块用来测试 较小代码块 的执行时间。

例如下面这个函数给数组的每一个元素加上固定值：

def add(arr, c):
    r = [0.] * len(arr)
    for i,v in enumerate(arr):
        r[i] = v + c
    return r

测试其运行时间：

import numpy as np
import timeit

arr = np.random.random(10**4).astype(np.float32)
sec: float = timeit.timeit(lambda: add(arr, 1), number=100)
print(sec)

timeit.timeit() 第一个参数可以是代码字符串，或者一个函数。

当执行对象是一个有参的函数时，可以使用匿名函数传参。

number 指定了运行代码运行次数，最后结果返回平均值。