如何在 python 中以像素数为 x 轴，灰度颜色为 y 轴 plot 图形？

Question

我下载了 MNIST 数据集和 plot 数据集的第一张图像，其中 imshow function 是灰度的。 这是我的代码。

import numpy as np
import matplotlib.pyplot as plt
import torch
import torchvision
from torchvision import transforms

# MNIST dataset 
transform = transforms.Compose([transforms.ToTensor()])
train_dataset = torchvision.datasets.MNIST(root='./data', 
                                           train=True, 
                                           download=True,
                                           transform=transform)

# Change data type : torch.Tensor -> numpy array
X = train_dataset.train_data.numpy()
y = train_dataset.train_labels.numpy()
print('Data size: ' + str(X.shape))

num_samples, height, width = X.shape 
dim = height * width

# Visualize MNIST data (1st digit)
plt.figure(1)
plt.imshow(X[0],cmap='gray_r')
plt.colorbar()
plt.title('1st image label: 5')

我得到了这个数字

将二维图像向量重塑为一维向量后，我想将 plot 变成这样的图。

x 轴表示像素数，y 轴表示其像素值作为灰度颜色，plot 就是这样，光谱之类的。 我知道如何重塑矢量，但我不知道如何制作 plot。 我可以在 matplotlib 模块中使用 function 使 plot 像第二个一样吗？ 还是在某个地方？

Answer 1

您是否正在寻找类似以下代码的内容。 我想象了mnist数字麻木

#load digits using the module load_digits()
digits = datasets.load_digits()

#visualize digits : 25 in 2D visualization
plt.figure()
plt.imshow(digits.images[25], cmap = plt.cm.gray_r, interpolation = 'nearest')
plt.show()

Output 是