将txt内容写入excel的python实现
最近搞深度学习的过程中,接到家中领导的科研任务需求,让把她的txt结果文件写入到excel中方便后续处理,于是写了个小脚本,不足之处欢迎交流。
txt数据格式如下:
import os
from xlrd import open_workbook
from xlutils.copy import copy
import xlwt
path = 'E:\\python_learning\\hymenoptera_data\\abc'
files = os.listdir(path) #读取指定路径下所有txt文件名
Total_reads = []
Aligned0 = []
Aligned1 = []
Aligned_greater = []
Overall_ali = []
#遍历txt文件
for file in files:
domain = os.path.abspath('E:\\python_learning\\hymenoptera_data\\abc')
file = os.path.join(domain, file)
fn = open(str(file))
f1 = fn.readlines()
Total_reads.append(f1[1].split()[2])
Aligned0.append(f1[2].split()[3] + f1[2].split()[4])
Aligned1.append(f1[3].split()[3] + f1[3].split()[4])
Aligned_greater.append(f1[4].split()[3] + f1[4].split()[4])
Overall_ali.append(f1[5].split()[3])
#写入excel
workbook = xlwt.Workbook(encoding='utf-8')
worksheet = workbook.add_sheet('0')
worksheet.write(0,0,'Total reads')
worksheet.write(0,1,'Aligned 0 time')
worksheet.write(0,2,'Aligned 1 time')
worksheet.write(0,3,'Aligned >1 times')
worksheet.write(0,4,'Overall alignment rate')
num = len(Total_reads)
for i in range(1,num+1):
worksheet.write(i,0,Total_reads[i-1])
worksheet.write(i,1,Aligned0[i-1])
worksheet.write(i, 2, Aligned1[i-1])
worksheet.write(i, 3, Aligned_greater[i-1])
worksheet.write(i, 4, Overall_ali[i-1])
workbook.save('data.xls')