numpy.expand_dims

numpy.expand_dims(a, axis)

Функция expand_dims() добавляет новую ось к массиву. Длинна новой оси всегда равна 1.

Параметры:
a - массив Numpy
Массив в который необходимо добавить новую ось.
axis - целое число
Положение новой оси.
Возвращает:
ndarray - массив NumPy
Массив в который добавленна новая ось.
Смотрите так же: squeeze, reshape, atleast_2d, atleast_3d


Примеры

>>> import numpy as np
>>> 
>>> a = np.array([1, 2, 3])
>>> 
>>> a.shape
(3,)
>>> 
>>> np.expand_dims(a, axis = 0)    #  Эквивалентно a[np.newaxis,:]
array([[1, 2, 3]])
>>> np.expand_dims(a, axis = 0).shape
(1, 3)
>>>
>>> np.expand_dims(a, axis = 1)    #  Эквивалентно a[:,np.newaxis]
array([[1],
       [2],
       [3]])
>>> np.expand_dims(a, axis = 1).shape
(3, 1)
>>> 
>>> 
>>> b = np.arange(6).reshape(3, 2)
>>> b
array([[0, 1],
       [2, 3],
       [4, 5]])
>>> b.shape
(3, 2)
>>> 
>>> np.expand_dims(b, axis = 0)
array([[[0, 1],
        [2, 3],
        [4, 5]]])
>>> np.expand_dims(b, axis = 0).shape
(1, 3, 2)
>>> 
>>> np.expand_dims(b, axis = 1)
array([[[0, 1]],

       [[2, 3]],

       [[4, 5]]])
>>> np.expand_dims(b, axis = 1).shape
(3, 1, 2)
>>> 
>>> np.expand_dims(b, axis = 2)
array([[[0],
        [1]],

       [[2],
        [3]],

       [[4],
        [5]]])
>>> np.expand_dims(b, axis = 2).shape
(3, 2, 1)