np.stack()

Объединяет массивы вдоль новой оси (увеличивает размерность).

Сигнатураnp.stack([a, b], axis=0)

В отличие от concatenate, добавляет новую ось.

import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
print(np.stack([a, b]))   # [[1 2 3]
                          #  [4 5 6]]