Combinations in Python

В Python для получения комбинаций элементов из списка можно использовать модуль itertools.

Например, если у нас есть список [1, 2, 3] и мы хотим получить все возможные комбинации длиной два, то можно использовать функцию combinations из модуля itertools:

python
import itertools
lst = [1, 2, 3]
comb = list(itertools.combinations(lst, 2))
print(comb)

Результатом будет список комбинаций: [(1, 2), (1, 3), (2, 3)].

Если же нам нужны комбинации с повторениями элементов, то можно использовать функцию combinations_with_replacement:

python
import itertools
lst = [1, 2, 3]
comb = list(itertools.combinations_with_replacement(lst, 2))
print(comb)

Результатом будет список комбинаций: [(1, 1), (1, 2), (1, 3), (2, 2), (2, 3), (3, 3)].

Модуль itertools также предоставляет другие полезные функции для работы с комбинациями, например, product (декартово произведение), permutations (перестановки) и т.д.