Puisque vous voulez trouver des doublons non ordonnés, la meilleure façon de faire est de transtyper. Typecast eux comme set . Puisque l'ensemble ne contient que immuable éléments. Donc, j'ai fait un ensemble de tuples .
Remarque :La meilleure façon d'éliminer les doublons est en faisant un set des éléments donnés.
>>> set(map(tuple,map(sorted,x)))
{(-3, -2, 4, 5), (-5, 0, 4, 5)}
Le meilleur moyen est de ne pas générer de doublons dès le départ.
L'idée est de créer d'abord toutes les combinaisons possibles de valeurs qui apparaissent plusieurs fois, où chacune apparaît 0, 1, ... fois. Ensuite, nous les complétons avec toutes les combinaisons possibles des éléments uniques.
from itertools import combinations, product, chain
from collections import Counter
nums = [-5,5,4,-3,0,0,4,-2]
def combinations_without_duplicates(nums, k):
counts = Counter(nums)
multiples = {val: count for val, count in counts.items() if count >= 2 }
uniques = set(counts) - set(multiples)
possible_multiples = [[[val]*i for i in range(count+1)] for val, count in multiples.items()]
multiples_part = (list(chain(*x)) for x in product(*possible_multiples))
# omit the ones that are too long
multiples_part = (lst for lst in multiples_part if len(lst) <= k)
# Would be at this point:
# [[], [0], [0, 0], [4], [4, 0], [4, 0, 0], [4, 4], [4, 4, 0], [4, 4, 0, 0]]
for m_part in multiples_part:
missing = k - len(m_part)
for c in combinations(uniques, missing):
yield m_part + list(c)
list(combinations_without_duplicates(nums, 4))
Sortie :
[[-3, -5, 5, -2],
[0, -3, -5, 5],
[0, -3, -5, -2],
[0, -3, 5, -2],
[0, -5, 5, -2],
[0, 0, -3, -5],
[0, 0, -3, 5],
[0, 0, -3, -2],
[0, 0, -5, 5],
[0, 0, -5, -2],
[0, 0, 5, -2],
[4, -3, -5, 5],
[4, -3, -5, -2],
[4, -3, 5, -2],
[4, -5, 5, -2],
[4, 0, -3, -5],
[4, 0, -3, 5],
[4, 0, -3, -2],
[4, 0, -5, 5],
[4, 0, -5, -2],
[4, 0, 5, -2],
[4, 0, 0, -3],
[4, 0, 0, -5],
[4, 0, 0, 5],
[4, 0, 0, -2],
[4, 4, -3, -5],
[4, 4, -3, 5],
[4, 4, -3, -2],
[4, 4, -5, 5],
[4, 4, -5, -2],
[4, 4, 5, -2],
[4, 4, 0, -3],
[4, 4, 0, -5],
[4, 4, 0, 5],
[4, 4, 0, -2],
[4, 4, 0, 0]]
