numpy.stack

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

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

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

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

Параметры:	arrays : последовательность array_like Каждый массив должен иметь одинаковую форму. axis : int, необязательно Ось в результирующем массиве, по которой объединяются входные массивы. out : ndarray, необязательно Если указано, место назначения для размещения результата. Форма должна быть правильной, соответствующей той, что вернула бы функция stack, если бы аргумент out не был указан.
Возвращает:	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]])