python二维数组输出为图像

最近参加kaggle-getstart级别的数字识别器比赛，通过以下代码，可以将一维数组变为二维数组，并输出为图像。

import pandas as pd
import numpy as np
import matplotlib.pyplot as pyplot

img_train = pd.read_csv('train.csv')#读取kaggle数据集，一维数组，代表28*28的图像
img_train = np.array(img_train)#这个是关键，转化为数组

trainSet = np.zeros((28, 28))
trainSet2 = np.zeros((1, 784))
trainSet2 =img_train[0][1:]
trainSet=trainSet2.reshape(28, 28)

print(trainSet)
pyplot.imshow(trainSet)
pyplot.show()

输出的结果是kaggle数据集的第一行代表的图像