Функция compress() модуля itertools в Python

Фильтр элементов на основе списка из False и True

Синтаксис:

import itertools

itertools.compress(data, selectors)

Параметры:

• data - последовательность элементов,
• selectors - последовательность, состоящая из значений 0, 1 или False, True.

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

• итератор.

Описание:

Функция compress() модуля itertools создает итератор, который фильтрует элементы из данных data, возвращая только те, которые имеют соответствующий элемент, равный True или 1 в последовательности selectors.

Функция itertools.compress() останавливается, когда исчерпана последовательность данных data или selectors.

Функция itertools.compress() примерно эквивалентна следующему коду:

def compress(data, selectors):
    # compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F
    return (d for d, s in zip(data, selectors) if s)

Примеры использования:

>>> from itertools import compress
>>> data = 'ABCDEF'
>>> selectors = [1,0,1,0,1,1]
>>> rez = compress(data, selectors)
>>> list(rez)
# ['A', 'C', 'E', 'F']