在项目中需要记录一下各模块的运行时间,自己简单写了个类
# RuntimeClass.py
# 记录运行时间类
import time
import datetime
import pandas as pd
class RuntimeClass:
def __init__(self):
self.StartTime=datetime.datetime.now() # 记录RuntimeClass开始运行时间
self.df=pd.DataFrame({'starttime': [], 'endtime': []})
def start(self,name): # 记录程序段开始运行时间
df_temp = pd.DataFrame({'starttime': [time.time()]}, index=[name])
self.df = pd.concat([self.df, df_temp], axis=0)
def end(self,name): # 记录程序段结束运行时间
self.df.loc[name,'endtime']=time.time()
def print(self):
print('RuntimeClass开始运行时间:'+str(self.StartTime))
for index, row in self.df.iterrows():
print(index+' 运行耗时 %f 秒' % (row['endtime'] - row['starttime']))
中间经历了DataFrame添加记录的麻烦,原来的append方法不行了,改了concat通过了,运行效果如下图