python 标注模块timeit: 测试函数的运行时间

在 Python 中,可以使用内置的 timeit 模块来测试函数的运行时间。timeit 模块提供了一个简单的接口来测量小段代码的执行时间。以下是使用 timeit 测试函数运行时间的一般步骤:

  1. 导入 timeit 模块:
import timeit
  1. 创建一个包含要测试的函数的语句块。可以使用 lambda 表达式来创建一个匿名函数。
  2. 使用 timeit.timeit() 函数来测量函数的运行时间。该函数接受四个参数:
  • 第一个参数是要测试的语句块或函数。
  • setup 参数是运行测试之前需要设置的语句块,例如导入模块或定义函数。
  • number 参数指定要执行测试语句的次数。
  • timer 参数用于指定计时器的类型,默认为系统默认的计时器。

例如:

# 测试函数运行时间
time_taken = timeit.timeit(stmt=lambda: my_function(), setup='from __main__ import my_function', number=1000)
print(f"函数运行时间: {time_taken} 秒")

这个例子中,stmt 参数使用了一个 lambda 表达式来调用 my_function(),setup 参数指定了导入函数的语句,number 参数设置了执行测试语句的次数。

需要注意的是,timeit 测试的是函数的纯运行时间,并不包括函数的调用开销和其他相关操作的时间。

另外,timeit 模块还提供了其他一些函数和工具,可以根据需要进行更精细的性能测试。详细的文档可以在 Python 官方文档中找到。

原文链接:,转发请注明来源!