python实现随机产生数据矩阵,将txt文件写入Excel中以及转置后写入Excel中

来源:互联网 发布:抗美援朝 知乎 编辑:程序博客网 时间:2024/06/05 19:56

    仅仅是因为用到这个东西就简单记录一下,做法很简单,这里先随机产生一个数据文件保存,之后将该数据文件内容写入到Excel中,数据文件可以形成一个矩阵,这里将矩阵转置后写入另一个Excel文件中,下面是具体的实现:


#!usr/bin/env python#encoding:utf-8'''__Author__:沂水寒城功能:将将txt文件写入Excel中以及转置后写入Excel中'''import randomfrom xlwt import *def genetate_random_nums(rows=100, cols=10, writefile='data.txt'):    '''    生成随机数    产生:100行,10列的数据文件    '''    matrix=[]    for i in range(rows):        tmp_list=[]        for j in range(cols):            tmp_list.append(random.randint(1,100))        matrix.append([str(o) for o in tmp_list])    with open(writefile, 'w') as f:        for one_list in matrix:            f.write(','.join(one_list)+'\n')def write_txt2_excel(sourcefile='data.txt', targetfile='vector.xls'):    '''    将txt中的内容写到Excel中    '''    with open(sourcefile) as f:        data_list=f.readlines()    j=0    book=Workbook(encoding = 'utf-8')    table=book.add_sheet('vector')    for one_list in data_list:        for i in range(len(one_list)):            table.write(j,i,one_list[i])        j+=1    book.save(targetfile)def write_txt2_excel_transfer(sourcefile='data.txt', targetfile='vector1.xls'):    '''    将txt中的内容转置后写到Excel中    '''    with open(sourcefile) as f:        data_list=f.readlines()    j=0    book=Workbook(encoding = 'utf-8')    table=book.add_sheet('vector1')    for one_list in data_list:        for i in range(len(one_list)):            table.write(i,j,one_list[i])        j+=1    book.save(targetfile)if __name__ == '__main__':    genetate_random_nums(rows=100, cols=10, writefile='data.txt')    write_txt2_excel(sourcefile='data.txt', targetfile='vector.xls')    write_txt2_excel_transfer(sourcefile='data.txt', targetfile='vector1.xls')



结果如下:

从上往下依次为:数据文件、Excel文件、转置后的Excel文件