mnist数据集读取和显示python

好的，下面是一个简单的示例代码，用于读取和显示MNIST数据集（需要先下载数据集文件）：

import numpy as np
import matplotlib.pyplot as plt

# 读取MNIST数据集
def read_mnist_data(filename):
    with open(filename, 'rb') as f:
        data = np.frombuffer(f.read(), np.uint8, offset=16)
    return data.reshape(-1, 28, 28)

# 显示MNIST数据集中的一个样本
def show_mnist_sample(data):
    plt.imshow(data, cmap='gray')
    plt.axis('off')
    plt.show()

# 测试代码
if __name__ == '__main__':
    train_data = read_mnist_data('train-images.idx3-ubyte')
    test_data = read_mnist_data('t10k-images.idx3-ubyte')
    show_mnist_sample(train_data[0])

这个示例代码使用numpy和matplotlib库，首先定义了一个函数用于读取MNIST数据集，这个函数会返回一个包含所有样本的numpy数组。然后定义了一个函数用于显示MNIST数据集中的一个样本，这个函数会显示一个灰度图像。最后，在测试代码中读取MNIST数据集文件，并显示其中的一个样本。