numpy.trim_zeros

numpy.trim_zeros(a, trim='fb')

Функция trim_zeros() удаляет нули из начала или конца указанного одномерного массива или последовательности.

Параметры:
a - одномерный массив или последовательность
Одномерный массив NumPy или любой объект который может быть преобразован в одномерный массив NumPy.
trim - строка: 'f', 'b' или 'fb' (необязательный)
Определяет где должны удаляться нули: 'f' - в начале массива, 'b' - в конце массива и 'fb' (по умолчанию) - в начале и конце массива.
Возвращает:
одномерный массив или последовательность
Одномерный массив с удаленными его из начала и (или) конца нулями. Тип данных массива сохраняется.
Смотрите так же: delete

Примеры

>>> import numpy as np
>>> 
>>> a = np.array([0, 0, 0, 1, 1, 1, 0, 0, 0])    #  Массив NumPy
>>> 
>>> np.trim_zeros(a)    #  Удаляет нули из начала и конца
array([1, 1, 1])
>>> 
>>> np.trim_zeros(a, 'f')    #  Удаляет нули только из начала
array([1, 1, 1, 0, 0, 0])
>>> 
>>> np.trim_zeros(a, 'b')    #  Удаляет нули только из конца
array([0, 0, 0, 1, 1, 1])
>>> 
>>> 
>>> #  Если входные данные это список или кортеж,
... #  то выходные данные будут также либо списком либо кортежем
... 
>>> b = [0, 0, 0, 1, 1, 1, 0, 0, 0]              #  Список
>>> c = (0, 0, 0, 1, 1, 1, 0, 0, 0)              #  Кортеж
>>>
>>> np.trim_zeros(b)
[1, 1, 1]
>>>
>>> np.trim_zeros(c)
(1, 1, 1)