python二维数组输出为图像

独钓寒江水

于 2018-01-28 12:19:28 发布

阅读量2.5w

点赞数 7

分类专栏： python学习

本文链接：https://blog.youkuaiyun.com/jsh458046505/article/details/79185682

版权

python学习专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种方法，用于将Kaggle数字识别竞赛的一维数组数据转换为二维图像并展示出来。该方法利用了Pandas、NumPy和Matplotlib等Python库。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近参加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数据集的第一行代表的图像