@Soul1

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

Функция product выдаёт сразу все комбинации, а можно ли сделать так, чтобы она выдавала уже отфильтрованные по условию комбинации ? Или может быть другая функция может это сделать ?
  • Вопрос задан
  • 67 просмотров
Пригласить эксперта
Ответы на вопрос 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')]
Ответ написан
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы