python计算程序所花时间装饰器
创建装饰器的文件(decorators.py)首先,我们创建一个名为decorators.py的文件,里面定义一个简单的装饰器,用于计算函数的执行时间。# decorators.py
import time
def timing_decorator(func):
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
print(f"{func.__name__} 执行时间: {end_time - start_time:.4f} 秒")
return result
return wrapper
引入并使用装饰器的文件(main.py)接着,我们在另一个文件main.py中导入上述定义的装饰器,并使用它来装饰一个函数。# main.py
from decorators import timing_decorator
@timing_decorator
def example_function(n):
sum = 0
for i in range(n):
sum += i
return sum
if __name__ in "__main__":
print(example_function(10000000))
在这个例子中,timing_decorator是一个装饰器,它测量被装饰函数的执行时间并打印出来。example_function是一个普通的函数,计算从0到n的累加和。通过在example_function前加上@timing_decorator,我们告诉Python在调用example_function时应用timing_decorator的功能。要运行这个示例,确保两个文件位于同一目录下,然后在命令行中运行python main.py,您将看到example_function的执行时间和结果。