python二维数组输出为图像
程序员文章站
2022-07-01 17:19:10
...
最近参加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数据集的第一行代表的图像