从Mnist读取数据,转为图片,将图片再转为数据存到excel里面

from PIL import Image

import numpy as np

import pandas as pd

 

train=pd.read_csv('\MnistData\train.csv')#读取Mnist的数据

t=np.reshape(train.iloc[:1,1:].values,(28,28))#转换为28,28格式,取第一行,以及第一列之后的数据,列1是标签行

im=Image.fromarray(t.astype(np.uint8))#将数据转换为图片

im.save('1.png')#保存图片

im.show()

 

im=Image.open('1.png')#打开图片

im=im.convert("L")#灰度

data=im.getdata()#获取图片数据

data=np.matrix(data,dtype='uint8').reshape(28,28)#转变形状为28,28

df=pd.DataFrame(data)#转变为dataframe

writer=pd.ExcelWriter('picture.xlsx')#创建一个excel书写器

df.to_excel(writer,'page1')#page1是sheet的名称

writer.save()

writer.close()

del im,data,df #释放内存

print("svae to excel suessfull")

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章