掌握Python中的time模块:测量和优化代码执行时间
import time
# 定义一个装饰器,用于测量函数执行时间
def timeit(func):
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
execution_time = end_time - start_time
print(f"执行函数 {func.__name__} 所需时间: {execution_time} 秒")
return result
return wrapper
# 使用装饰器测量函数执行时间
@timeit
def long_running_function():
# 模拟长时间运行的操作,例如数据处理或计算
time.sleep(2) # 休眠2秒来模拟耗时操作
return "完成"
# 调用被装饰的函数
long_running_function()
这段代码定义了一个名为timeit
的装饰器,它会在函数调用前后记录时间,并计算出函数执行所需的时间。使用这个装饰器,可以轻松地测量任何需要的函数执行时间。在这个例子中,long_running_function
模拟了一个耗时的操作,如数据处理或计算密集型任务。当调用这个函数时,它会输出函数执行所需的时间。
评论已关闭