numpy.stack

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

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

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

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

Параметры

arrayssequence of array_like

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

axisint, optional

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

outndarray, optional

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

Возвращает

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