将mnist获得的数据还原成图片形式

MNIST是一个手写数字数据集，里面所包含的数据元素是类似于一个1×784矩阵和1×10矩阵的结构，那么，如何将这些数据元素转化为更加直观的图像呢？通过以下python代码，可以实现。

 from PIL import Image
 from tensorflow.examples.tutorials.mnist import input_data
 import numpy as np
 import matplotlib.pyplot as plt

 #获取数据集
 mnist = input_data.read_data_sets("MNIST_data/",one_hot=True)
 #fdata为图像数据，label为该图像的标签
 fdata,label = mnist.train.next_batch(1)
 #将1*784矩阵变为28*28的矩阵
 data = np.matrix(fdata).reshape(28,28)
 data = data*255
 #矩阵转图像
 new_im = Image.fromarray(data.astype(np.uint8))
 new_im.show()

效果如下：