python计算程序所花时间装饰器

2024-04-29  本文已影响0人  DD丿

创建装饰器的文件(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的执行时间和结果。

上一篇 下一篇

猜你喜欢

热点阅读