如何在多维数组中获取正确的数组

Question

Python编程： x_train.shape是（60000,28,28,1） 我如何将数据转换为（60000,28,28） 当我输入x_train [0]时，它可以显示28 x 28的数组。

Answer 1

您可以使用numpy.squeeze()从数组形状中删除单个一维条目。

np.squeeze(x_train)

Link to documentation