使用裝飾器記錄函數體運行時間：

import time import wraptclass Timing:timings = {}enabled = Falsedef __init__(self, enabled=True):Timing.enabled = enableddef __str__(self):return "Timing"__repr__ = __str__@classmethoddef timeit(cls, level=0, func_name=None, cls_name=None, prefix="[Method] "):@wrapt.decoratordef wrapper(func, instance, args, kwargs):if not cls.enabled:return func(*args, **kwargs)if instance is not None:instance_name = "{:>18s}".format(instance.__class__.__name__)else:instance_name = " " * 18 if cls_name is None else "{:>18s}".format(cls_name)_prefix = "{:>26s}".format(prefix)try:_func_name = "{:>28}".format(func.__name__ if func_name is None else func_name)except AttributeError:str_func = str(func)_at_idx = str_func.rfind("at")_dot_idx = str_func.rfind(".", None, _at_idx)_func_name = "{:>28}".format(str_func[_dot_idx+1:_at_idx-1])_name = instance_name + _prefix + _func_name_t = time.time()rs = func(*args, **kwargs)_t = time.time() - _ttry:cls.timings[_name]["timing"] += _tcls.timings[_name]["call_time"] += 1except KeyError:cls.timings[_name] = {"level": level,"timing": _t,"call_time": 1}return rsreturn wrapper@classmethoddef show_timing_log(cls, level=2):print()print("=" * 110 + "\n" + "Timing log\n" + "-" * 110)if cls.timings:for key in sorted(cls.timings.keys()):timing_info = cls.timings[key]if level >= timing_info["level"]:print("{:<42s} : {:12.7} s (Call Time: {:6d})".format(key, timing_info["timing"], timing_info["call_time"]))print("-" * 110)@classmethoddef disable(cls):cls.enabled = False

使用和測試：

if __name__ == '__main__':class Test:timing = Timing()def __init__(self, rate):self.rate = rate@timing.timeit()def test(self, cost=0.1, epoch=3):for _ in range(epoch):self._test(cost * self.rate)@timing.timeit(prefix="[Core] ")def _test(self, cost):time.sleep(cost)class Test1(Test):def __init__(self):Test.__init__(self, 1)class Test2(Test):def __init__(self):Test.__init__(self, 2)class Test3(Test):def __init__(self):Test.__init__(self, 3)test1 = Test1()test2 = Test2()test3 = Test3()test1.test()test2.test()test3.test()test1.timing.show_timing_log()runfile('/root/2half/Util/Timing.py', wdir='/root/2half/Util') () ============================================================================================================== Timing log --------------------------------------------------------------------------------------------------------------Test1 [Core] _test : 0.3006141 s (Call Time: 3)Test1 [Method] test : 0.3007839 s (Call Time: 1)Test2 [Core] _test : 0.6010602 s (Call Time: 3)Test2 [Method] test : 0.6012352 s (Call Time: 1)Test3 [Core] _test : 0.9015043 s (Call Time: 3)Test3 [Method] test : 0.901685 s (Call Time: 1) --------------------------------------------------------------------------------------------------------------

總結

以上是生活随笔為你收集整理的Python学习：装饰器使用，timeit()记录程序运行在哪里，耗时多少的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。