Python:将 1×28×28 数组转换为 28×28 矩阵

Question

我有一个存储为 1×28×28 数组 (uint8) 的灰度图像，我希望使用 matplotlib 显示它。

相关代码如下:

import matplotlib.pyplot as plt

image = get_image()
print image.shape    # prints (1, 28, 28)
plt.imshow(image)

此代码导致 TypeError:图像数据的尺寸无效。

为了与 imshow() 兼容，我需要将数组压缩为 28×28(删除冗余维度)或将 2D 矩阵(28×28)复制到3 (RGB) 或 4 (RGBA) channel 。

我倾向于使用第一种方法，但是 squeeze() 不会删除冗余维度。

如何将 1×28×28 数组转换为 28×28 矩阵？

Answer 1

你可以通过 squeeze 或 reshape:

 imshow(np.squeeze(image))


 imshow(image.reshape((28,28))