numpy.stack

numpy.stack(arrays, axis=0) [source]

Объединение последовательности массивов вдоль новой оси.

Параметр axis определяет индекс новой оси в размерностях результата. Например, если axis=0 он будет первой осью, а если axis=-1 — последней осью.

Новая в версии 1.10.0.

Параметры:	arrays : последовательность array_like Каждый массив должен иметь одинаковую форму. axis : int, необязательно Ось в результирующем массиве, вдоль которой объединяются входные массивы.
Возвращает:	stacked : ndarray Массив со сложенными элементами имеет на одну размерность больше, чем входные массивы.

См. также

concatenate

Объединение последовательности массивов вдоль существующей оси.

split

Разделение массива на список подмассивов равного размера.

Примеры

>>> arrays = [np.random.randn(3, 4) for _ in range(10)]
>>> np.stack(arrays, axis=0).shape
(10, 3, 4)

>>> np.stack(arrays, axis=1).shape
(3, 10, 4)

>>> np.stack(arrays, axis=2).shape
(3, 4, 10)

>>> a = np.array([1, 2, 3])
>>> b = np.array([2, 3, 4])
>>> np.stack((a, b))
array([[1, 2, 3],
       [2, 3, 4]])

>>> np.stack((a, b), axis=-1)
array([[1, 2],
       [2, 3],
       [3, 4]])