Библиотека Python itertools содержит много полезных функций.

Одна из них — compress(), которая возвращает итератор по элементам из data, для которых соответствующий элемент в selectors равен True.

Вот пример: