python - 强制timeit自动选择语句执行次数

Question

在命令行中，我可以做:

python -m timeit 'a = 1'

根据文档:

If -n is not given, a suitable number of loops is calculated by trying successive powers of 10 until the total time is at least 0.2 seconds.

这很好用，但是在我的程序中使用 timeit 时如何获得相同的行为？如果我省略 timeit.timeit 调用的 number 参数，它将简单地默认为 1000000。

Answer 1

文档没有明确表示存在这样的功能，我假设不存在。幸运的是，为您定义一个包装器并不难:

import timeit

def auto_timeit(stmt='pass', setup='pass'):
    n = 1
    t = timeit.timeit(stmt, setup, number=n)

    while t < 0.2:
        n *= 10
        t = timeit.timeit(stmt, setup, number=n)

    return t / n # normalise to time-per-run