Python装饰器实现函数运行时间的计算

Python装饰器实现函数运行时间的计算

目录

个人理解

例子:调用函数的同时对函数进行计时

实现方法1:@语法糖

代码:

实现结果:

实现方法2:闭包

代码:

实现结果:

实现方式1和 2的差异

总结

个人理解

装饰器: 通过闭包和将一个函数作为另一个函数参数的形式,实现已有功能的灵活调用

例如:

首先设置了一个time_master的计时器函数,在运行某个函数的同时,用来统计函数的耗时

那么,如果想知道函数性能, 每次写完新的函数后,都放到time_master函数中运行一次来统计。

——> 这是比较麻烦的。相当于虽然有了一个称,但是所有的新买食材都需要逐个放到称上去称一遍,来获得食材的重量

——> 如果能够省略掉一次一次上称这个步骤就好了

——> 比如每个食材进门的同时就从称上走过,那进来的同时,也就有了重量的数据

例子:调用函数的同时对函数进行计时 实现方法1:@语法糖 代码: # 涉及到计时,需要引入time模块 import time # 定义一个对函数运行耗时统计的计时器,用func作为形参来代替需要统计的函数 def time_master(func): def call_func(): print("计时器开始:函数开始调用:") start_time = time.time() func() end_time = time.time() print('计时器结束,函数调用完成') return print(f'计时器结果返回:函数调用耗时{end_time-start_time:.2f}') return call_func @time_master # 即在调用myfunc函数时,不是直接调用myfunc # 而是将myfunc函数作为一个参数放入到@的装饰器中,然后去调用装饰器 def myfunc(): time.sleep(2) print('myfunc函数运行') time.sleep(4) print('myfunc函数运行结束') myfunc() # 调用myfunc 实现结果:

计时器开始:函数开始调用:
myfunc函数运行
myfunc函数运行结束
计时器结束,函数调用完成
计时器结果返回:函数调用耗时6.01

实现方法2:闭包

可以理解成,本来定义了一个myfunc的函数,但这个函数本身没有计时的功能,而恰巧有一个time_master的函数,在运行子函数的同时,还会对子函数进行计时

因此,通过重新定义 myfunc = time_master(myfunc), 即将myfunc作为参数传入到time_master中,作为myfunc函数的新定义

代码: import time def time_master(func): def call_func(): print("计时器开始:函数开始调用:") start_time = time.time() func() end_time = time.time() print('计时器结束,函数调用完成') print(f'计时器结果返回:函数调用耗时{end_time-start_time:.2f}') return call_func def myfunc(): time.sleep(2) print('myfunc函数运行') time.sleep(4) print('myfunc函数运行结束') myfunc = time_master(myfunc) # 和实现方法一的差距就在于是在myfunc前面@装饰器,还是在后面对myfunc函数进行二次定义 myfunc() 实现结果:

计时器开始:函数开始调用:
myfunc函数运行
myfunc函数运行结束
计时器结束,函数调用完成
计时器结果返回:函数调用耗时6.01

实现方式1和 2的差异

实现方式1和实现方式2: 其实没什么区别,无非是一开始就用@time_master来规定,还是定义完myfunc之后,再多做一步让myfun放入time_master中去执行

总结

本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注易知道(ezd.cc)的更多内容!               

推荐阅读