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

Question

我下载了 MNIST 数据集并使用 imshow 函数以灰度绘制数据集的第一张图像。 这是我的代码。

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')

我得到了这个数字

将二维图像向量重塑为一维向量后，我想绘制 像这样的图。

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

Answer 1

您是否正在寻找类似以下代码的内容。我可视化mnist digit numb

#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()

输出是