Объединить многомерный массив Numpy с массивом 1D 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).

На основе эта ссылка, я пробовал:

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

Однако в результате получается следующая форма: массив ([124800, 784, 124800, 1]).

Как можно Я вставляю y_train позади X_train, чтобы форма была (124800, 785)?

Answer 1

Если вы объединяете два массива, вы должны объединять данные внутри массивов, а не форму. Кроме того, вы хотите объединить по второй («короткой») оси, которая равна axis=1:

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