将形状从（X，1）转换为（X）

Question

现有的train_labels，具有以下属性

('labels_train shape ', (3000,))
('type of labels_train ', <type 'numpy.ndarray'>)

，另一个数组是Y，它具有以下属性

('Y ', (3000,1))
('type of Y ', <type 'numpy.ndarray'>)

如何将Y分配给labels_train，或如何Y具有与labels_train相同的形状？

Answer 1

您可以使用squeeze功能：

Y = Y.squeeze()

这将删除单身维度，以便Y.shape为(3000,)。

Answer 2

如果您要将内容从Y复制到labels_train，则以下内容应该有效。

labels_train[:] = Y[:,0]

如果您想要将Y重塑为与labels_train具有相同的形状（仅当它们具有相同数量的元素时才有效）。

Y = Y.reshape(labels_train.shape)