numpy.stack

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

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

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

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

Параметры:	arrays : последовательность array_like Каждый массив должен иметь одинаковую форму. axis : int, необязательно Ось в результирующем массиве, вдоль которой объединяются входные массивы.
Возвращает:	stacked : ndarray Массив, полученный в результате объединения, имеет на одну размерность больше, чем входные массивы.

См. также

concatenate

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

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