J'ai des difficultés à créer une liste/tableau 2D

Question

J'ai de la difficulté à créer une liste 2D de permutations. Voici un code minimal pour reproduire le problème

class Solution: 
def permute(self, A): 
    A = sorted(A) 
    print A 
    A_out = [] 
    A_out.append(A) 
    for iter0 in range(4): 
     A[0] = A[0] + 1 
     print A 
     A_out.append(A) 
    return A_out 

sol = Solution() 
A = [1, 2, 3] 
print sol.permute(A) 

Pour cette entrée particulière (1, 2, 3) la sortie est

[1, 2, 3] 
[2, 2, 3] 
[3, 2, 3] 
[4, 2, 3] 
[5, 2, 3] 
[[5, 2, 3], [5, 2, 3], [5, 2, 3], [5, 2, 3], [5, 2, 3]] 

mais la sortie désirée est

[1, 2, 3] 
[2, 2, 3] 
[3, 2, 3] 
[4, 2, 3] 
[5, 2, 3] 
[[1, 2, 3], [2, 2, 3], [3, 2, 3], [4, 2, 3], [5, 2, 3]] 

I pense qu'il a quelque chose à faire de copie profonde/chose de copie superficielle mais je ne suis pas sûr de savoir comment corriger cela car je ne suis pas très familier avec Python. Comment puis-je le réparer?

Answer 1

Il s'agit en effet de copies superficielles. La liste A que vous continuez d'ajouter fait toujours référence à la même valeur. Ceci est lié à la mutabilité des listes en Python.

Vous devez ajouter une nouvelle copie de la liste chaque fois pour les rendre indépendants l'un de l'autre. Vous pouvez le faire en utilisant l'opérateur de tranche [:] car il crée une nouvelle copie de la liste. Ainsi, vous pouvez simplement l'utiliser lorsque vous appelez append

def permute(self, A): 
    A = sorted(A) 
    print A 
    A_out = [] 
    A_out.append(A[:]) 
    while (self.checkVal(A) != -1) : 
     A = self.nextState(A,self.checkVal(A)) 
     print A 
     A_out.append(A[:]) 
    return A_out