Я хотел бы создать перестановки итерации, в том числе для подмножеств итерации, без дубликатов или пустого набора:
(1,2,3) --> (
(1,),
(2,),
(3,),
(1, 2),
(1, 3),
(2, 3),
(1, 2, 3),
(1, 3, 2),
(2, 1, 3),
(2, 3, 1),
(3, 1, 2),
(3, 2, 1)
)
По сути, это функция powerset
, предоставляемая more-itertools в сочетании с itertools.permutations
:
from itertools import combinations, permutations
def powerset_permutations(iterable):
"([1,2]) --> (1,) (2,) (1,2) (2,1)"
s = list(iterable)
for r in range(1, len(s)):
yield from combinations(s, r)
yield from permutations(s)
>>> list(powerset_permutations([1, 2, 3]))
[(1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)]
Чтобы включить пустой набор, измените range(1, len(s))
на range(len(s))
:
def powerset_permutations(iterable):
"([1,2]) --> (1,) (2,) (1,2) (2,1)"
s = list(iterable)
for r in range(len(s)):
yield from combinations(s, r)
yield from permutations(s)
>>> list(powerset_permutations([1,2,3]))
[(), (1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)]