numpy.expand_dims
numpy.expand_dims(a, axis)
Функция expand_dims() добавляет новую ось к массиву. Длинна новой оси всегда равна 1.
-
- a - массив Numpy
- Массив в который необходимо добавить новую ось.
- axis - целое число
- Положение новой оси.
-
- ndarray - массив NumPy
- Массив в который добавленна новая ось.
Примеры
>>> 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)