最近在运行一段程序时需要多次统计程序运行时间,并加以优化。想来也比较简单,但是每次都来一波时间统计时,就没必要次次都写一遍,自然就想到用很Pythonic的装饰器咯,一劳永逸嘛,下面直接上代码。
import time
def timer(func):
def call_func(*args, **kwargs):
print("计时开始")
start_time = time.time()
func(*args, **kwargs)
end_time = time.time()
total_time = end_time - start_time
print("计时结束")
print(f"程序用时{int(total_time // 60)}分{total_time % 60:.2f}秒")
return call_func
if __name__ == '__main__':
@timer
def test(second): # 测试函数
time.sleep(second)
test(3.4)
下面是运行结果
我们先实现一个通用的装饰器,并在func函数前后内置的time模块实现计时统计,并用python3.6推出的新特性f字符串(超级推荐,又快又直观,还简便)实现打印结果。
当我们需要统计某个程序或代码块的运行时长时(如上述test函数),在test函数上加上我们刚写的@timer装饰器就可以了。
当然我们需要复用这个模块,这时只需要将这个mytimer.py(以上代码块的名字,自己可以随意起别的)拷贝到我们的项目里面,导入该模块并在需要用到计时的地方用装饰器修饰就可以了,下面是一个实现的例子:
# 导入计时器装饰器
from mytimer import timer
@timer # 给函数加装饰器
def spark_demo(usetime):
"""
假设我们要统计运行spark中spark_demo的总时长
"""
import time
time.sleep(usetime)
pass
if __name__ == '__main__':
spark_demo(74.4)
运行结果:
只要需要用到计时的地方加一个@timer,简简单单,舒舒服服就搞定了!这就是pythoner头发发量惊人的多的原因!!!
谁能仅靠十年寒窗苦读,去经营百岁人生?
活到老,学到老!
知识改变人生!
共勉!