minist数据集图片查看

minist数据集图片查看
网上很多代码都是from tensorflow.examples.tutorials.mnist import input_data，是tensorflow早期版本的，无法运行，一堆错误，想要把tensorflow版本退回去又容易和numpy产生冲突等，很烦！！
我的tensorflow是2.4.1版本，一般采用keras 方法加载数据集
首先从网上下载mnist.npz文件，存到data_path = 'F:\minist\data_set_tf2\mnist.npz’路径下。
运行下面的代码就行了

import matplotlib.pyplot as plt
from tensorflow.keras import datasets
data_path = 'F:\minist\data_set_tf2\mnist.npz'
(train_X, train_Y), (test_X,test_Y) = datasets.mnist.load_data(path=data_path)
fig, ax = plt.subplots(nrows=4,ncols=5,sharex='all',sharey='all')##4行5列
ax = ax.flatten()  # ax变成长度为20的数组
for i in range(20):
    img = train_X[i].reshape(28, 28)
    ax[i].imshow(img,cmap='Greys')
ax[0].set_xticks([])
ax[0].set_yticks([])
plt.tight_layout()
plt.show()