Mnist 数据 Python 的图像 - 显示图像时出错

Question

我正在使用 Mnist 数据集，以了解机器学习，目前我正尝试将 Mnist 数据集中的第一个数字显示为图像，但遇到了问题。

我有一个尺寸为 784x10000 的矩阵，其中每一列都是数据集中的一个数字。我自己创建了矩阵，因为 Mnist 数据集以文本文件的形式出现，这本身给我带来了很多问题，但这本身就是一个问题。

下面的 MN_train 矩阵是我的 784x10000 大矩阵。所以我在下面要做的是填充一个 28x28 矩阵，以显示我的图像。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

grey = np.zeros(shape=(28,28))
k = 0

for l in range(28):
    for p in range(28):
        grey[p,l]=MN_train[k,0]
        k = k + 1
print grey
plt.show(grey) 

但是当我尝试显示图像时，出现以下错误：

具有多个元素的数组的真值是不明确的。使用 a.any() 或 a.all()

后面是一个看起来不像数字五的图像，正如我所期望的那样。

是否有什么我忽略了，或者这是否告诉我为了构造 MN_train 矩阵而对文本文件的操作导致了错误？

Answer 1

您得到的错误是因为您将数组提供给show。 show 只接受一个布尔参数 hold=True 或 False。
为了创建图像图，您需要使用imshow。

plt.imshow(grey)
plt.show()  # <- no argument here

还要注意循环效率很低。您可以只重塑输入列数组。

完整的代码如下所示

import numpy as np
import matplotlib.pyplot as plt

MN_train = np.loadtxt( ... )

grey = MN_train[:,0].reshape((28,28))

plt.imshow(grey)
plt.show()