unike.utils.Timer 源代码
# coding:utf-8
#
# unike/utils/Timer.py
#
# created by LuYF-Lemon-love <luyanfeng_nlp@qq.com> on July 6, 2023
# updated by LuYF-Lemon-love <luyanfeng_nlp@qq.com> on Feb 24, 2023
#
# 该脚本定义了计时器类.
"""
:py:class:`Timer` - 计时器类。
"""
import time
[文档]class Timer:
"""记录多次实验的时间。
通过调用 :py:meth:`stop` 能够返回距离上一次 :py:meth:`stop` 调用或创建 :py:class:`Timer` 对象时的时间间隔。
:py:meth:`avg` 能够返回多次实验的平均时间；:py:meth:`sum` 能够返回多次实验的总时间。"""
[文档] def __init__(self):
"""创建 Timer 对象。"""
#: 存放时间间隔的列表
self.times: list[float] = []
#: 记录当前时间
self.current: float = None
#: 记录上一次的时间
self.last: float = None
self.__restart()
def __restart(self):
"""重启计时器。"""
self.last = self.current = time.time()
[文档] def stop(self) -> float:
"""停止计时器并将时间记录在列表中。
:returns: 返回最后一次的间隔时间。
:rtype: float
"""
self.current = time.time()
self.times.append(self.current - self.last)
self.last = self.current
return self.times[-1]
[文档] def avg(self) -> float:
"""返回平均时间。
:returns: 平均时间
:rtype: float
"""
return sum(self.times) / len(self.times)
[文档] def sum(self) -> float:
"""返回时间总和。
:returns: 时间总和。
:rtype: float
"""
return sum(self.times)