【星光05】Python 全局时间节点计时器 GlobalTimer

这是一个记录时间节点和代码运行顺序的计时器。可以在代码的任何地方调用它来记录运行时间并保存到本地文件中。

✨代码部分

创建timer.py文件,创建GlobalTimer类。

import time as t
from typing import Dict, List

import pandas as pd


class GlobalTimer:
    """
    全局计时器,跨文件实现代码的时间节点记录。
    """
    __time_dict: Dict[str, List[float or None]] = {}  # 时间存储字典
    time_slice_count: int = 0  # 时间切片计数

    def __init__(self):
        pass

    def timing(self, name: str, func, para: tuple):
        """
        测量某一函数的运行时间。
        :param name: 时间片段名称
        :param func: 函数
        :param para: 函数输入参数,元组类型
        :return: 函数的返回值
        """
        # 计时部分
        start_time = t.time()
        value_return = func(*para)
        end_time = t.time()
        # 记录时间
        time = end_time - start_time
        self.add_value(name, time)
        # 返回函数输出
        return value_return

    def add_value(self, name: str, time: float):
        """
        添加时间到时间存储字典,并增加时间切片数值。
        :param name: 时间片段名称
        :param time: 时间
        """
        # 判断是否存在记录,如果不存在,则新建历史记录
        if name not in self.__time_dict.keys():
            self.__time_dict[name] = [None for _ in range(self.time_slice_count)]
        # 添加新的记录
        for key, value in self.__time_dict.items():
            if key == name:
                value.append(time)
            else:
                value.append(None)
        # 时间切片加1
        self.time_slice_count += 1

    def get_value(self):
        return self.__time_dict

    def save_local(self, path):
        pd.DataFrame(self.__time_dict).to_csv(path)

    def __str__(self):
        return str(self.__time_dict)

✨使用部分

注意:所有示例均使用匿名函数,实际不推荐使用。

示例1:

from timer import GlobalTimer

if __name__ == '__main__':
    global_timer = GlobalTimer()

    result = global_timer.timing(name="test_time",
                                 func=lambda x, y: x + y,
                                 para=(1, 1))

    print(f"func result: {result}, time: {global_timer}")

结果:

func result: 2, time: {'test_time': [0.0]}

示例2:

from timer import GlobalTimer

if __name__ == '__main__':
    global_timer = GlobalTimer()
    global_timer.timing(name="test1_time",
                        func=lambda x, y: x + y,
                        para=(1, 1))
    global_timer.timing(name="test2_time",
                        func=lambda x, y: x + y,
                        para=(1, 1))
    global_timer.timing(name="test3_time",
                        func=lambda x, y: x + y,
                        para=(1, 1))
    global_timer.timing(name="test2_time",
                        func=lambda x, y: x + y,
                        para=(1, 1))
    global_timer.save_local("./time.csv")

结果:

time.csv

示例3:

# 对于只有一个输入参数的 func 需要注意元组类型的写法,以下方法任选其一。

# 推荐
global_timer.timing(name="test_time",
                    func=lambda x: x,
                    para=(1,))

# 或者
global_timer.timing(name="test_time",
                    func=lambda x: x,
                    para=tuple([1]))

🍀破碎的想法🍀
Hello米娜桑,这里是英国留学中的杨丝儿。我的博客的关键词集中在编程、算法、机器人、人工智能、数学等等,点个关注吧,持续高质量输出中。
🌸唠嗑QQ群:兔叽的魔术工房(942848525)
⭐️B站账号:YangSierCode000(活跃于知识区生活区和动画区)

【星光05】Python 全局时间节点计时器 GlobalTimer

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2022年3月22日
下一篇 2022年3月22日

相关推荐