Можно ли сделать фильтр для функции product в itertools?

Question

[Soul1]

Функция product выдаёт сразу все комбинации, а можно ли сделать так, чтобы она выдавала уже отфильтрованные по условию комбинации ? Или может быть другая функция может это сделать ?

Comments

[aRegius]

Пример?

Answer 1

[We1tkinD]

Как вариант можно использовать filter и lambda.

>>>from itertools import product

>>> list(product('ABCD',  'xy'))
[('A', 'x'), ('A', 'y'), ('B', 'x'), ('B', 'y'), ('C', 'x'), ('C', 'y'), ('D', 'x'), ('D', 'y')]

>>>list(filter(lambda x: x[0] in ('A', 'B'), product('ABCD', 'xy')))
[('A', 'x'), ('A', 'y'), ('B', 'x'), ('B', 'y')]

Comments

[Soul1]

Тут вопрос, создаст ли функция product временный список, содержащий все возможные комбинации. Если создаст, тогда вариант не годится, так как этот список всю память сожрёт и ещё не хватит.

[We1tkinD]

Soul1, и product, и filter возвращают специальные объекты, которые по сути генераторы. То есть временный список не создаётся.

[javedimka]

Soul1, Чтобы отфильтровать данные по какому-либо условию - их всё равно придётся создать.
Вопрос в том, будут ли они где-то храниться. Ответ - нет. функция product() это генератор, который возвращает значения по требованию, но, если аналогичный генератор подать ей на вход - она пожрёт его сразу и будет хранить в памяти, поэтому, если ты передаёшь ей на вход генератор, который читает значения из 8 гигабайтного файла - все эти 8 гб попадут в память. Является ли функция фильтр генератором - не знаю

[Soul1]

Вариант с filter не помог, память всё так же переполняется.