mnist数据集读取与显示用numpy

读取MNIST数据集可以使用Python的`numpy`和`gzip`库。以下是一个读取MNIST数据集的示例代码：

import numpy as np
import gzip

def load_mnist(path, kind='train'):
    """Load MNIST data from `path`"""
    labels_path = f'{path}/{kind}-labels-idx1-ubyte.gz'
    images_path = f'{path}/{kind}-images-idx3-ubyte.gz'
    
    with gzip.open(labels_path, 'rb') as lbpath:
        labels = np.frombuffer(lbpath.read(), dtype=np.uint8, offset=8)
        
    with gzip.open(images_path, 'rb') as imgpath:
        images = np.frombuffer(imgpath.read(), dtype=np.uint8, offset=16).reshape(len(labels), 784)
        
    return images, labels

其中，`path`是MNIST数据集在本地的目录，`kind`是数据集的类型（训练集或测试集）。该函数返回两个`numpy`数组，分别是图像数据和标签数据。

要显示MNIST图像，可以使用`matplotlib`库。以下是一个将MNIST图像显示为灰度图像的示例代码：

import matplotlib.pyplot as plt

def plot_mnist(images, labels):
    """Plot MNIST images"""
    fig, axs = plt.subplots(2, 5, figsize=(10, 5))
    axs = axs.flatten()
    for i in range(10):
        img = images[labels == i][0].reshape(28, 28)
        axs[i].imshow(img, cmap='gray')
        axs[i].set_title(str(i))
        axs[i].axis('off')
    plt.show()

该函数将MNIST图像绘制为一个2行5列的子图，每个子图显示一个数字。可以使用以下代码调用函数并显示MNIST图像：

images, labels = load_mnist('path/to/mnist', kind='train')
plot_mnist(images, labels)