numpy.stack

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

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

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

Добавлена в версии 1.10.0.

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