使用matplotlib.pyplot.imshow()在pycharm中无法绘图

最新推荐文章于 2024-07-13 13:16:07 发布

SUN_QQS

最新推荐文章于 2024-07-13 13:16:07 发布

阅读量768

点赞数

分类专栏： Python 文章标签： python matplotlib

本文链接：https://blog.csdn.net/QQSqqs_/article/details/110306241

版权

Python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

问题： 此种方式使用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()