为了方便以后查找,记录下平时用到的一些常用方法:
功能:比较两个TXT文件每行的指定位置
import os import time def compute_ap(input,output): with open(input,'r') as f_in: len_f = len(f_in.readlines()) print("compute_ap len_f:",len_f) with open(input,'r+') as f_in: line_1 = f_in.readlines() with open(output,'r+') as f_out: line_2 = f_out.readlines() for row in range(len_f-1): line_1_data = line_1[row].strip('\r\n').split(',') line_2_data = line_1[row].strip('\r\n').split(',') if(abs(float(line_1_data[-2])-float(line_2_data[-1]))): count_temp =count_temp + 1 f_in.close() f_out.close()
功能:读取txt文件,每次一条放入指定txt文件,并执行推理
import os,time import ac_inference def extract_line_txt(input_path): with open(input_path,'r') as f: len_lines = len(f.readlines()) with open(input_path,'r') as f: line = f.readlines() for row in range(len_lines-1): datatxt=[] line_data = line[row].strip('\r\n').split(',') datatxt.append(line_data[0]) datatxt.append(line_data[1]) datatxt.append(line_data[2]) datatxt.append(line_data[3]) input_data = ac_inference.test_input_data(datatxt) ac_inference.inference(input_data) f.close()
功能:读取excel文件,每次一条放入指定txt文件,并执行推理
import xlrd,xlwt,random import os,time import ac_inference #把时间8:30转换成8.5 def transformtime(timestr): timelst = timestr.split(":") time = int(timelst[0]+int(timelst[1])/60 return str(time) def extract_line_xls(filepath): wb=xlrd.open_workbook(filepath) table=wb.sheet_by_index(0) rowsnum = table.nrows-1 for row in range(rowsnum): datatxt=[] data=table.row_values(row+1) print('extract_line_data[0]:",data[0]) datatxt.append(transformtime(data[1]) datatxt.append(data[2]) print('datatxt:',datatxt) input_data=ac_inference.test_input_data(datatxt) ac_inference.inference(input_data)input_path =os.path.join(os.getcwd(),'ac_data_input.txt')output_path = os.path.join(os.getcwd(),'ac_output.txt')