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)