Python obtém todas as permutações de números
-
20-09-2019 - |
Pergunta
Estou tentando exibir todas as permutações possíveis de uma lista de números, por exemplo, se eu tiver 334, quero obter:
3 3 4
3 4 3
4 3 3
Preciso fazer isso para qualquer conjunto de dígitos de até 12 dígitos.
Tenho certeza de que é provavelmente bastante simples usar algo como o itertools.combinações, mas não consigo acertar a sintaxe.
Tia Sam
Solução
>>> lst = [3, 3, 4]
>>> import itertools
>>> set(itertools.permutations(lst))
{(3, 4, 3), (3, 3, 4), (4, 3, 3)}
Outras dicas
sem itertools
def permute(LIST):
length=len(LIST)
if length <= 1:
yield LIST
else:
for n in range(0,length):
for end in permute( LIST[:n] + LIST[n+1:] ):
yield [ LIST[n] ] + end
for x in permute(["3","3","4"]):
print x
resultado
$ ./python.py
['3', '3', '4']
['3', '4', '3']
['3', '3', '4']
['3', '4', '3']
['4', '3', '3']
['4', '3', '3']
Você quer permutações, não combinações. Ver: Como gerar todas as permutações de uma lista em Python
>>> from itertools import permutations
>>> [a for a in permutations([3,3,4])]
[(3, 3, 4), (3, 4, 3), (3, 3, 4), (3, 4, 3), (4, 3, 3), (4, 3, 3)]
Observe que está permitindo os dois 3 (que é a coisa correta matematicamente fazer), mas não é o mesmo que o seu exemplo. Isso só fará a diferença se houver números duplicados na sua lista.
Eu usaria o Python itertools
, mas se você tiver que implementar isso você mesmo, aqui está o código que retorna todas as permutações de um tamanho especificado para uma lista de valores.
Exemplo: values = [1,2,3]
, size = 2
=> [[3, 2], [2, 3], [2, 1], [3, 1], [1, 3], [1, 2]]
def permutate(values, size):
return map(lambda p: [values[i] for i in p], permutate_positions(len(values), size))
def permutate_positions(n, size):
if (n==1):
return [[n]]
unique = []
for p in map(lambda perm: perm[:size], [ p[:i-1] + [n-1] + p[i-1:] for p in permutate_positions(n-1, size) for i in range(1, n+1) ]):
if p not in unique:
unique.append(p)
return unique