import xlwt
import os
import numpy as np
def read_txt(path):
filelist = os.listdir(path)
file_names = np.array([file for file in filelist if file.endswith('.txt')], dtype=object)
return file_names
def transform(sheet, sheet_row, path):
for i in range(len(sheet_row)):
sheet.write(0, i, sheet_row[i])
paths = path + '/' + sheet_row[i]
f = open(paths)
index = 1
for line in f:
data = line
sheet.write(index, i, data)
index += 1
return sheet
workbook = xlwt.Workbook()
path1 = 'E:/LinJN_bishe/暴雨数据/上中下游/上/predict'
path2 = 'E:/LinJN_bishe/暴雨数据/上中下游/上/standard_39'
path3 = 'E:/LinJN_bishe/暴雨数据/上中下游/上/gsmap_gauge_39'
path4 = 'E:/LinJN_bishe/暴雨数据/上中下游/上/station_39'
#创建第一个sheet
sheet1 = workbook.add_sheet('predict', cell_overwrite_ok=True)
sheet1_row = read_txt(path1)
sheet1 = transform(sheet1,sheet1_row,path1)
# for i in range(len(sheet1_row)):
# sheet1.write(0, i, sheet1_row[i])
# paths = path1 + '/' +sheet1_row[i]
# f = open(paths)
# index = 1
# for line in f:
# data = line
# sheet1.write(index,i, data)
# index += 1
#创建第二个sheet
sheet2 = workbook.add_sheet('standard_39', cell_overwrite_ok=True)
sheet2_row = read_txt(path2)
sheet2 = transform(sheet2,sheet2_row,path2)
# for i in range(len(sheet2_row)):
# sheet2.write(0, i, sheet2_row[i])
# paths = path2 + '/' + sheet2_row[i]
# f = open(paths)
# index = 1
# for line in f:
# data = line
# sheet2.write(index,i, data)
# index += 1
#创建第三个sheet
sheet3 = workbook.add_sheet('gsmap_gauge_39', cell_overwrite_ok=True)
sheet3_row = read_txt(path3)
sheet3 = transform(sheet3,sheet3_row,path3)
# for i in range(len(sheet3_row)):
# sheet3.write(0, i, sheet3_row[i])
# paths = path3 + '/' + sheet3_row[i]
# f = open(paths)
# index = 1
# for line in f:
# data = line
# sheet3.write(index,i, data)
# index += 1
# 创建第四个sheet
sheet4 = workbook.add_sheet('station_39', cell_overwrite_ok=True)
sheet4_row = read_txt(path4)
sheet4 = transform(sheet4,sheet4_row,path4)
# for i in range(len(sheet4_row)):
# sheet4.write(0, i, sheet4_row[i])
# paths = path4 + '/' + sheet4_row[i]
# f = open(paths)
# index = 1
# for line in f:
# data = line
# sheet4.write(index, i, data)
# index += 1
workbook.save('E:/LinJN_bishe/暴雨数据/上中下游/上/data.xls')