python如何将图片的像素矩阵绘制成图片（python,matplotlib):TypeError: Invalid shape (1, 28, 28) for image data

矩阵变成图片，这个问题使用（python , matplotlib ) 可以轻松实现。

import matplotlib.pyplot as plt
#使用格式
plt.imshow(x)#其中x为图片的像素矩阵。

其中的x为：

array-like or PIL image
Supported array shapes are:
(M, N): an image with scalar data. The values are mapped to colors using normalization and a colormap. See parameters norm, cmap, vmin, vmax.
(M, N, 3): an image with RGB values (0-1 float or 0-255 int).
(M, N, 4): an image with RGBA values (0-1 float or 0-255 int), i.e. including transparency.
The first two dimensions (M, N) define the rows and columns of the image.
Out-of-range RGB(A) values are clipped.

从上面我们可以看到，其只支持的矩阵形式为：

(M, N)
(M, N, 3)
(M, N, 4)

所以如果我们要把手写数字识别中的图片(1,28,28)矩阵绘制成图片，需要把其压缩一个维度(28,28)。

否则报错：

TypeError: Invalid shape (1, 28, 28) for image data

正确的应该如下：

print(img.shape)
plt.imshow(img.squeeze(0))#把0维度去掉。