Python 使用 MNIST 数据集显示手写数字图片

本教程将指导您使用 Python 读取已下载的 MNIST 数据集,并显示随机手写数字图片。我们将使用 NumPy 和 Matplotlib 库完成此任务。

1. 导入必要的库

首先,需要先导入必要的库,包括 NumPy 和 Matplotlib:

import numpy as np
import matplotlib.pyplot as plt

2. 读取 MNIST 数据集

然后,读取 MNIST 数据集,可以使用 NumPy 的 load 函数。假设您的 MNIST 数据集文件名为 'mnist.npz',并存储在 'D:/dataset/MNIST' 文件夹中:

data = np.load('D:/dataset/MNIST/mnist.npz')

3. 获取图片和标签

接着,将数据集中的图片和标签分别存储到变量 x 和 y 中:

x = data['x_train']
y = data['y_train']

4. 显示随机图片

现在就可以随机选择一张图片并显示出来,可以使用 Matplotlib 的 imshow 函数:

# 随机选择一张图片
index = np.random.randint(len(x))
image = x[index]

# 显示图片
plt.imshow(image, cmap='gray')
plt.show()

5. 完整代码

import numpy as np
import matplotlib.pyplot as plt

# 读取 MNIST 数据集
data = np.load('D:/dataset/MNIST/mnist.npz')

# 获取图片和标签
x = data['x_train']
y = data['y_train']

# 随机选择一张图片
index = np.random.randint(len(x))
image = x[index]

# 显示图片
plt.imshow(image, cmap='gray')
plt.show()

注意:

  • 请确保您已下载 MNIST 数据集并将其放置在 'D:/dataset/MNIST' 文件夹中。
  • 'mnist.npz' 文件名可能与您的实际文件名不同,请根据您的情况进行调整。
  • 如果您没有安装 NumPy 和 Matplotlib 库,请使用 pip 命令进行安装:
pip install numpy matplotlib
Python 使用 MNIST 数据集显示手写数字图片

原文地址: https://www.cveoy.top/t/topic/npjH 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录