matplotlib.pyplot 可视化MNIST
时间: 2025-01-22 17:03:27 浏览: 37
### 使用 `matplotlib.pyplot` 对 MNIST 数据集进行可视化
为了展示如何使用 `matplotlib.pyplot` 来可视化 MNIST 数据集中的手写数字图片,下面提供了一段 Python 代码示例。这段代码会加载 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()
# 创建一个新的图形窗口用于绘图
plt.figure(figsize=(10, 5))
# 绘制前9张训练集中图片及其标签
for i in range(9):
plt.subplot(3, 3, i + 1)
plt.imshow(x_train[i], cmap='gray')
plt.title(f'Label: {y_train[i]}')
plt.axis('off')
plt.tight_layout()
plt.show()
```
上述代码首先导入必要的库,并通过 Keras 接口获取了 MNIST 数据集[^1]。接着创建了一个大小为 10×5 英寸的新图表,在这个图表上按照 3 行 3 列布局排列九幅子图来呈现部分训练样例。每一张子图都设置了灰度色彩映射 (`cmap='gray'`) 和关闭坐标轴线以便更清晰地查看图像本身。最后调用了 `tight_layout()` 函数自动调整各元素间距使整个画面更加紧凑美观。
阅读全文
相关推荐



















