Есть ли способ объединить два массива с разными размерами?

Question

Я работаю с моделью глубокого обучения, которая пытается объединить метку с размерами (1,2) с массивом (25,25).Я не совсем уверен, возможно ли получить измерение (627,0), однако в сводке модели говорится, что это ожидаемая форма ввода.

Я пытался объединить их, но я получаю сообщение об ошибке "все размеры входного массива, кроме оси объединения, должны точно соответствовать", как и ожидалось.

    x = np.concatenate((X[1], to_categorical(Y_train[1]))   

Где X = (25,25) и Y_train равен (1,0), делая to_categorical (Y_train [1]) равным (2,1).

Есть ли способ получить это (627, 0) измерение с этимиразмеры?

Answer 1

@ Psidom имеет отличный ответ на этот вопрос:

Допустим, у вас есть 1-й и 2-й массив

Вы можете использовать numpy.column_stack:

np.column_stack((array_1, array_2))

Что неявно преобразует 1-й массив в 2-й, и, таким образом, эквивалентен np.concatenate((array_1, array_2[:,None]), axis=1).

---

a = np.arange(6).reshape(2,3)
b = np.arange(2)

a
#array([[0, 1, 2],
#       [3, 4, 5]])

b
#array([0, 1])

np.column_stack((a, b))
#array([[0, 1, 2, 0],
#       [3, 4, 5, 1]])