问题: 此种方式使用matplotlib.pyplot.imshow()无法绘图
import numpy
import matplotlib.pyplot as plt
。。。省略了很多代码,完整代码在下面
image_array= numpy.asfarray(all_values[1:]).reshape((28, 28))
plt.imshow(image_array, cmap='Greys', interpolation='None') # 绘图面板不显示
解决办法:
- 导入pylab包
- 在matplotlib.pyplot.imshow() 方法后面添加 pylab.show()方法
import numpy
import matplotlib.pyplot as plt
# 直接使用plt.imshow无法显示图片,需要导入pylab包
import pylab
# 读取训练集
def readFile():
# 打开文件
data_file = open("./mnist_dataset/mnist_train_100.csv","r")
# readline()一次读一行会更有效率,但是我们的文件小,可以使用readlines()将文件一次读入内存
data_list = data_file.readlines()
# 关闭文件
data_file.close()
return data_list
# 绘图
def show():
# 接收读取出的数据集
data_list = readFile()
# 读取第一行,并以逗号分隔
all_values = data_list[0].split(',')
# 绘制
# numpy.asfarray() 将文本字符串转换称实数,并创建这些数字的数组
# .reshape((28,28)) 确保数字列表每28个元素折返一次,形成28*28的像素数组
image_array= numpy.asfarray(all_values[1:]).reshape((28, 28))
# 使用 matplotlib.pyplot 的imshow()函数绘出image_array像素数组
# cmap='Greys' 灰度调色板
plt.imshow(image_array, cmap='Greys', interpolation='None')
pylab.show()
# 注意是 == ,判断
if __name__ == '__main__':
show()
本代码用于绘制手写数字,数据集可以从mnist数据集下载中下载mnist_train_100.csv文件,感兴趣的可以一试。