Python自定义装饰器-统计函数的耗时和CPU使用率

最新推荐文章于 2024-07-28 14:34:24 发布

原创

最新推荐文章于 2024-07-28 14:34:24 发布 · 278 阅读

0 ·

CC 4.0 BY-SA版权

本文为博主原创文章，未经博主允许不得转载。

文章标签：

#python #性能优化

经常遇到耗时分析、资源利用率分析等，最近手搓了一份统计函数的耗时和CPU使用率的Python装饰器，纯个人的自定义用法，后续再改进。

import time
import psutil
import numpy as np


def profile(func):
    def wrapper(*args, **kwargs):
        result = None
        elapsed, cpu_percentage, cpu_nozero_times = 0.0, 0.0, 0
        psp = psutil.Process(os.getpid())
        t0 = time.time()
        psp.cpu_percent()  # last call
        result = func(*args, **kwargs)
        cpu_percent = psp