numpy.expand_dims

numpy.expand_dims(a, axis) [source]

Расширение формы массива.

Вставка новой оси, которая будет расположена в позиции axis в расширенной форме массива.

Примечание

Прежде чем NumPy 1.13.0, ни axis < -a.ndim - 1 ни axis > a.ndim не генерировали ошибки или не размещали новую ось так, как задокументировано. Эти значения осей теперь устарели и в будущем будут генерировать ошибку AxisError.

Параметры:	a : array_like Входной массив. axis : int Позиция в расширенных осях, где размещается новая ось.
Возвращает:	res : ndarray Массив вывода. Число измерений на единицу больше, чем у входного массива.

См. также

squeeze

Обратная операция, удаление одиночных измерений

reshape

Вставка, удаление и объединение измерений, а также изменение размера существующих

doc.indexing, atleast_1d, atleast_2d, atleast_3d

Примеры

>>> x = np.array([1,2])
>>> x.shape
(2,)

Следующее эквивалентно x[np.newaxis,:] или x[np.newaxis]:

>>> y = np.expand_dims(x, axis=0)
>>> y
array([[1, 2]])
>>> y.shape
(1, 2)

>>> y = np.expand_dims(x, axis=1)  # Equivalent to x[:,newaxis]
>>> y
array([[1],
       [2]])
>>> y.shape
(2, 1)

Обратите внимание, что некоторые примеры могут использовать None вместо np.newaxis. Это одни и те же объекты:

>>> np.newaxis is None
True