Permutations and Combinations in Python (itertools)
published at 2023-05-28
python
algorithm
data-structure
itertools
permutations-combinations
programming-language
Extracted from itertools — Functions creating iterators for efficient looping — Python 3.11.3 documentation
PERMUTATIONS
itertools.permutations(iterable, r=None)
returns successive r
length permutations of elements in the iterable.
If r
is not specified or is None
, then r
defaults to the length of the iterable and all possible full-length permutations are generated.
Ex.1.
from itertools import permutations
a = [1, 2, 3]
for p in permutations(a):
print(p)
Output:
(1, 2, 3)
(1, 3, 2)
(2, 1, 3)
(2, 3, 1)
(3, 1, 2)
(3, 2, 1)
Ex.2.
from itertools import permutations
a = ['A', 'B', 'C', 'D']
for p in permutations(a, 2):
print(p)
Output:
('A', 'B')
('A', 'C')
('A', 'D')
('B', 'A')
('B', 'C')
('B', 'D')
('C', 'A')
('C', 'B')
('C', 'D')
('D', 'A')
('D', 'B')
('D', 'C')
Learn more here
COMBINATIONS
Notice: r
is NOT optional. (Not like permutations
)
WITH NO REPLACEMENTS
itertools.combinations(iterable, r)
returns r
length subsequences of elements from the input iterable.
Ex.
from itertools import combinations
a = [1, 2, 3]
for c in combinations(a, 2):
print(c)
Output:
(1, 2)
(1, 3)
(2, 3)
Learn more here
WITH REPLACEMENTS
itertools.combinations_with_replacement(iterable, r)
returns r
length subsequences of elements from the input iterable allowing individual elements to be repeated more than once.
Ex.
from itertools import combinations_with_replacement
a = [1, 2, 3]
for c_with_replacement in combinations_with_replacement(a, 2):
print(c_with_replacement)
Output:
(1, 1)
(1, 2)
(1, 3)
(2, 2)
(2, 3)
(3, 3)
Learn more here