Python 性能分析：timeit 指南

timeit 模块是 Python 内置的一个用于测量小段代码运行时间的库，非常适合用于性能分析和基准测试。

快速开始

要使用 timeit 模块，首先需要导入它：

import timeit

然后，你可以定义一个要测试的代码片段，并使用 timeit 库来执行：

execution_time = timeit.timeit('some_code', globals=globals(), number=1000)

这里，some_code 是你想要测试的代码字符串，globals() 函数确保了代码片段能够访问全局命名空间中的变量，而 number 参数指定了代码需要执行的次数。

选项详解

timeit 模块提供了许多选项，以下是一些常用的：

setup: 一个字符串，表示在测试之前执行的代码，通常用于设置测试环境。
number: 代码需要执行的次数，默认是 1,000,000。
interval: 测试之间的最小间隔时间，默认是 0.01 秒。

例如：

execution_time = timeit.timeit('some_code', setup='import my_module', number=1000, interval=0.1)

示例

以下是一个使用 timeit 模块的示例，比较两种不同循环方式的性能：

import timeit

# 使用 for 循环
for_loop_time = timeit.timeit('for i in range(10000): pass', number=1000)

# 使用 while 循环
while_loop_time = timeit.timeit('i = 0; while i < 10000: i += 1', number=1000)

print(f'For loop time: {for_loop_time}')
print(f'While loop time: {while_loop_time}')

扩展阅读

想要了解更多关于 timeit 模块的信息，可以阅读官方文档：Python timeit 模块