Discussion :

[Neitsa]

Bonjour,

[Python 3.x]

Problème:

Je cherche à générer toutes les permutations d'un nombre donné:

Code python :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
 
import itertools
 
def get_all_perms(n):
    n_str = str(n)
    perms = [int(''.join(x)) for x in list(itertools.permutations(n_str))]
    return perms
 
if __name__ == '__main__':
    perms = get_all_perms(123)
    print(perms)

Output:

[123, 132, 213, 231, 312, 321]

Maintenant avec n=122

[122, 122, 212, 221, 212, 221]

Alors que je cherche plutôt à obtenir (donc: liste sans doublon):

[122, 212, 221]

Ce qui fonctionne avec ça:

Code python :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
 
def get_all_perms(n):
    n_str = str(n)
    perms = [int(''.join(x)) for x in list(itertools.permutations(n_str))]
    out_perms = list()
    for perm in perms:
        if perm not in out_perms:
            out_perms.append(perm)
    return out_perms

Mais ce dernier code me donne l’impression d'être suboptimal (deux listes utilisées, un parcours pour chaque liste, etc.).

Est-il possible , à l'intérieur d'une 'list comprehension' d'ajouter une condition pour éviter qu'un élément soit ajouter s'il existe déjà dans la liste ?

Quelqu'un aurait-il une idée pour un code propre concernant ce problème?

Merci beaucoup

[josmiley]

si l'ordre n'a pas d'importance : set()

[tyrtamos]

Bonjour,

Avec Python 2.7 et 3.x, on peut construire un "set comprehension" qui permet de tout faire d'un coup:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
import itertools
 
n = 122
perm = list({int(''.join(x)) for x in itertools.permutations(str(n))})
print perm
[122, 212, 221]

Le "set comprehension" est fait par l'accolade et empêche les doublons sans autre test.

Tyrtamos

[Neitsa]

Bonjour,

Merci beaucoup pour vos réponses et la solution ! Je n'avais pas pensé au 'set'...