将二维数据写入Excel表格当中。
import xlwt
import numpy
import pandas
def w_excel(datas,file_name,column_name):
'''
:param datas:二位数据矩阵
:param file_name:不带后缀的文件名字
:param column_name:数据列名,必须与数据的列数保持一直
:return:一个文件
'''
if type(datas) is not numpy.ndarray: #判断数据是否为numpy,如果不是转化为numpy
datas = numpy.array(datas)
if datas.dtype != "numpy.float64": #判断数据是否为numpy.float64 如果不是,转化为float64
datas = datas.astype(numpy.float64)
workbook = xlwt.Workbook(encoding='utf-8') #设置文件编码
sheet = workbook.add_sheet(file_name) #设置文件sheet表名
for col, column_data in enumerate(column_name): #写入列名
sheet.write(0, col, column_data)
for row, data in enumerate(datas): #写入数据
for col, column_data in enumerate(data):
sheet.write(row+1, col, column_data)
workbook.save(file_name+'.xls') #保存文件到统计目录下
a = numpy.random.random((63,4)).astype(numpy.float32)
w_excel(a,"babbabab",["a","b","c","d"])
将excel表格中的数据读入numpy数组中:
import pandas
path = './babbabab.xls' # excel表格地址
data2 = pandas.read_excel(path).values #读取excel表格内容,注意默认第一行为列名,不读入作为数据