为了方便以后查找,记录下平时用到的一些常用方法:
功能:比较两个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')