使用timeit模块可以度量函数的运行时间。timeit模块提供了一个Timer类,可以用来测量代码片段的执行时间。下面是使用timeit来度量函数运行时间的步骤:
import timeit
语句来导入。timer = timeit.Timer(stmt='函数调用语句', setup='导入函数所需的语句')
来创建计时器对象。其中,stmt参数是要度量时间的函数调用语句,setup参数是导入函数所需的语句。timeit()
方法来执行计时器,该方法会返回函数运行的时间。可以使用result = timer.timeit(number=执行次数)
来执行计时器,其中number参数是执行函数的次数。print("函数运行时间:", result)
来输出函数的运行时间。下面是一个示例代码,演示如何使用timeit来度量函数的运行时间:
import timeit
def my_function():
# 函数代码
# 创建计时器对象
timer = timeit.Timer(stmt='my_function()', setup='from __main__ import my_function')
# 执行计时器,执行1000次函数调用,并输出结果
result = timer.timeit(number=1000)
print("函数运行时间:", result)
在上面的示例中,首先导入了timeit模块,然后定义了一个名为my_function的函数。接下来,创建了一个计时器对象timer,使用了my_function作为要度量时间的函数调用语句。最后,执行了计时器1000次,并将结果输出。
注意:在使用timeit度量函数运行时间时,需要确保函数的调用语句和导入语句在setup参数中正确设置,以保证计时器能够正确执行函数。
领取专属 10元无门槛券
手把手带您无忧上云