numpy.stack

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

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

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

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

Parameters

arrayssequence of array_like

Каждый массив должен иметь одинаковую форму.

axisint, optional

Ось в результирующем массиве, вдоль которой объединяются входные массивы.

outndarray, optional

Если указано, место назначения для размещения результата. Форма должна быть правильной, соответствующей той, что вернул бы stack, если бы аргумент out не был указан.

Returns

stackedndarray

Массив со стеком имеет на одну размерность больше, чем входные массивы.

См. также

concatenate

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

block

Сборка nd-массива из вложенных списков блоков.

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