Timeit.timeit 方法。哪种方式是正确的：内部还是外部？

Question

我想检查每个功能的速度，但遇到了问题。如您所见，下面的代码包含两个相同的函数：func & func2。但是 func 是直接在 timeit.timeit 方法中声明的。 经过测试，我得到了一个结果：Func result: 0.08, func 2 result: 0.21.。这是一个巨大的差异。

这里有一个问题：为什么相同的功能有不同的速度？从全局范围获取 func2 需要时间吗？

import timeit


def func2() -> range:
    return range(1024)


def main():
    """
    Number of seconds for each function to perform with 500k executions
    """
    func_res = round(timeit.timeit(
        'def func() -> range: return range(1024); func()',
        number=500_000
        ),
    2)
    func2_res = round(timeit.timeit(
        'func2()',
        number=500_000,
        globals=globals()
        ),
    2)
    return f'Func result: {func_res}, func 2 result: {func2_res}.'

if __name__ == '__main__':
    print(main())

附注'def func() -> range: return range(1024)' ---> 'def func() -> range: return range(1024); func()' 已被编辑。

Answer 1

您正在测试的两个语句完全不同。

第一个 (def func() -> range: return range(1024); func()) 仅测试 Python 在使用主体 return range(1024); func() 声明函数时的速度。第二个实际上调用了该函数。

由于timeit 也可以接受可调用对象，因此您可以这样做

timeit.timeit(
    func2,
    number=500_000,
)

无论如何。

让我们尝试一个更好的测试平台：

import timeit


def func2() -> range:
    return range(1024)


def t(name, stmt, setup="pass"):
    res = timeit.timeit(stmt, setup, number=5_000_000, globals=globals()) * 1000
    print(f"{name:>25} {int(res):5d}")


def main():
    t("declare a func", "def func() -> range: return range(1024); func()")
    t("declare & run func", "def func() -> range: return range(1024)\nfunc()")
    t("run local func", "func()", "def func() -> range: return range(1024)")
    t("run global func2 by name", "func2()")
    t("run func2 by ref", func2)


if __name__ == "__main__":
    main()

你可以看到声明func的选项和调用它也慢得多（我机器上的数字，YMMV）：

           declare a func   547  # <-- doesn't call range!
       declare & run func  1282  # <-- slow...
           run local func   854  # <-- declare function once & call it
 run global func2 by name   836
         run func2 by ref   774