Манипуляции с массивом NumPy: функция squeeze ()

функция numpy.squeeze ()

Функция squeeze () используется для удаления одномерных записей из формы массива.

Синтаксис:

 numpy.squeeze (a, axis = None) 

Версия: 1.15.0

Параметр:

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

squeezed [ndarray] Входной массив, но со всеми или подмножеством измерений длины 1 удалены. Это всегда само по себе или взгляд в.

Повышает: ValueError - Если ось не None, а сжатая ось не имеет длины 1

Пример 1: функция numpy.squeeze ()

>>> import numpy as np
>>> a = np.array([[[0], [2], [4]]])
>>> a.shape
(1, 3, 1)
>>> np.squeeze(a).shape
(3,)

Иллюстрированная презентация:

Пример 2: функция numpy.squeeze ()

>>> import numpy as np
>>> a = np.array([[[0], [2], [4]]])
>>> np.squeeze(a, axis=0).shape
(3, 1)
>>> np.squeeze(a, axis=1).shape
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.6/dist-packages/numpy/core/fromnumeric.py", line 1201, in squeeze
    return squeeze(axis=axis)
ValueError: cannot select an axis to squeeze out which has size not equal to one
>>> np.squeeze(a, axis=2).shape
(1, 3)

Python - NumPy Code Editor:

Предыдущий: expand_dims ()
Далее: Изменение вида массива asarray ()