用matplotlib显示mnist手写数据集
时间: 2024-09-17 13:05:29 浏览: 39
mnist手写数字识别数据集npz文件.zip
Matplotlib是一个广泛用于Python的数据可视化库,可以用来展示MNIST手写数字数据集。MNIST是一个常用的手写数字图像数据集,包含了60,000张训练图片和10,000张测试图片,每张图片都是28x28像素的灰度图像。
以下是使用matplotlib显示MNIST数据集中一张图片的基本步骤:
```python
import matplotlib.pyplot as plt
from tensorflow.keras.datasets import mnist
# 加载MNIST数据集
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 选择一张图片并调整其尺寸以便于显示
index = 5 # 你可以修改这个索引来查看不同的图片
img = x_train[index].reshape(28, 28)
# 显示图片
plt.imshow(img, cmap='gray') # 使用灰色映射
plt.title('Label: %i' % y_train[index]) # 标注图片标签
plt.axis('off') # 关闭坐标轴
plt.show()
```
运行这段代码后,你会看到一张MNIST手写的数字及其相应的标签。
阅读全文