将多维 Numpy 数组与一维 numpy 数组连接起来

Question

我有两个 numpy 数组（来自 EMNIST 数据集的字母）：

import scipy .io
emnist = scipy.io.loadmat(DATA_DIR + '/emnist-letters.mat')
data = emnist ['dataset']
X_train = data ['train'][0, 0]['images'][0, 0]
y_train = data ['train'][0, 0]['labels'][0, 0]

具有以下尺寸：

X_train.shape = (124800, 784)

y_train.shape = (124800, 1)

现在，我想将两者连接起来，这样新的形状将是：(124800, 785)。

基于this链接，我试过了：

np.concatenate((X_train.shape, y_train.shape), axis = 0)

但是，这会导致以下形状：array([124800, 784, 124800, 1])。

如何在X_train 后面“粘贴”y_train 以使形状变为 (124800, 785)？

Answer 1

如果连接两个数组，则必须连接数组内的数据，而不是形状。此外，您想在第二个（“短”）轴上连接，即axis=1：

np.concatenate((X_train, y_train), axis=1)