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

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

block

Сборка массивов из блоков.

Примеры

>>> 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]])