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: Invalid dimensions for image data。

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

我的偏好是使用第一种方法，但是squeeze() 不会删除冗余维度。

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

Answer 1

您可以通过挤压或重塑：

 imshow(np.squeeze(image))


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

Answer 2

image = image[0]

很简单。