timeit

介绍

在Python中，我们经常需要评估代码的性能，查看哪些实现方式更有效率。timeit模块是Python标准库中的一个工具，可以用来测量代码段的执行时间。

使用示例

让我们看一个简单的示例，比较两个列表生成式的执行时间：

import timeit

# 列表生成式方式1
list_comp_1 = timeit.timeit('[i for i in range(1000)]', number=1000)

# 列表生成式方式2
list_comp_2 = timeit.timeit('list(range(1000))', number=1000)

print("列表生成式1执行时间：", list_comp_1)
print("列表生成式2执行时间：", list_comp_2)

timeit.timeit是timeit模块中的一个函数，它通过多次执行同一段代码来计算平均运行时间，以减少误差。

具体用法

timeit模块不仅可以计算单行代码的执行时间，还可以用来测试多行代码的执行时间。除此之外，它还可以用来测试函数的执行时间。

def test_function():
    return [i for i in range(1000)]

function_time = timeit.timeit('test_function()', globals=globals(), number=1000)
print("函数执行时间：", function_time)

在函数调用时，我们可以使用globals=globals()参数将当前全局命名空间传递给timeit，这样测试函数的性能时就不会受到外部变量的影响。

结论

通过timeit模块，我们可以快速、准确地测量Python代码的执行时间，帮助我们选择最佳的实现方式并优化程序性能。