不同之处:
保存内容为,以log命名的文件夹内存有一个txt文件+一张对应图像
import os
import argparse
import numpy as np
import pandas as pd
import matplotlib
import matplotlib.pyplot as plt
def main():
#####################命令行获取输入输出文件路径############################
parser = argparse.ArgumentParser("Demo of argparse")
parser.add_argument('log_path', type=str,
default=" ", help="path to log file")
parser.add_argument('out_path',type=str,
default=" ",help ="path to result file")
args = parser.parse_args()
log_path = args.log_path
out_path=args.out_path
current_path = os.path.abspath(log_path)
current_outpath = os.path.abspath(out_path)
print("输入文件路径为: {}".format(current_path))
print("输出结果文件保存在: {}".format(current_outpath))
files = os.listdir(log_path)
os.mkdir(out_path)
#os.mkdir(out_path+'\\figures')
##########################批量读取#################################
for file_name in files:
fname,fename=os.path.splitext(file_name)
os.mkdir(out_path+'\\'+fname)
print(fname)
input = open(log_path+"\\"+file_name, 'r')
fpsNumArr = []
coutArr=[]
inttimeArr=[]
for line in input:
line = line.split()
if 'fps->' in line:
#print(line)
line[-1]=line[-1].lstrip('"')
fpsNumArr.append(int(line[-1]