numpy.stack

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

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

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

Новая функция в версии 1.10.0.

Параметры

arrayssequence of array_like

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

axisint, optional

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

outndarray, optional

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

Возвращаемое значение

stackedndarray

Массив, полученный путём объединения, имеет на одну размерность больше, чем входные массивы.

См. также

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