numpy.stack

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

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

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

Введено в версии 1.10.0.

Параметры

arraysпоследовательность array_like

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

axisint, необязательно

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

outndarray, необязательно

Если указано, место назначения для размещения результата. Форма должна быть правильной, совпадая с той, которую вернул бы 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([4, 5, 6])
>>> np.stack((a, b))
array([[1, 2, 3],
       [4, 5, 6]])

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