Python: Como copiar um array/list de forma segura (Explicação completa)

Muito cuidado quando for copiar um array/list em python, pois você pode provocar um grande problema no seu código.

Se você simplesmente copiar atribuir a variável de um array para outra variável, as alterações que você fizer na segunda variável serão realizadas também na primeira. Veja um exemplo:

a = [1,2,3]
b = a
b[0] = 4

Você, pode até achar que alterou só o array da variável b, mas na verdade o array da variável a também foi alterado. Veja a execução desse código imprimindo o valor das variáveis a e b.

Perceba que a alteração no array da variável b (b[0] = 4) teve impacto também no array da variável a.

Existe várias formas de copiar um array de forma segura em python, onde a alteração em um não impacta no outro e eu vou mostrar 5 delas mais a frente.

Mas se quiser uma resposta rápida sobre como fazer essa cópia basta usar a função copy() do array, assim:

b = a.copy()

Entretanto acho que você merece uma explicação mais aprofundada sobre o que acontece por baixo dos panos.

O que acontece por baixo dos panos

Vamos pegar o nosso exemplo acima para entender o que acontece quando criamos um array.

Quando criamos um array e atribuimos à uma variável, por exemplo a = [1,2,3], o python cria um objeto na memória e atribui o endereço desse objeto à variável. Pensa em algo assim:

Quando copiamos uma variável para outra, o que estamos copiando é o valor da variável, nesse caso o endereço.

Então, quando fazemo a atribuição b = a, estamos copiando o valor da variável a para a variável b.

Mais ou menos assim.

Ou seja, copiamos endereço para o mesmo objeto array!

Isso significa que se fizermos uma alteração no objeto array em qualquer uma das variáveis, estaremos alterando o mesmo objeto. Por isso acontece o exemplo que mostrei no inicio.

Então, o que precisamos fazer é uma cópia do objeto array e atribuir o endereço desse novo objeto para a outra variável. Mais ou menos assim:

Dessa forma as alterações feitas no objeto de uma das variáveis não impacta o outro objeto.

Existem várias formas de fazer isso, como o método copy().

4 formas de copiar um objeto array

Como já disse a forma mais simples de copiar um objeto array é através do método copy.

1 - Método copy() da classe Array

Perceba no exemplo abaixo que fizemos a cópia do objeto array da variável a e atribuimos esse novo objeto à variável b (b = a.copy()) e alteração b[0] = 4 não alterou o array de a, somente o array de b.

2 - Fazendo um filtro trazendo todos os itens de um array com [:]

Uma outra ideia é fazer um filtro no array original trazendo todos os registros assim: b = a[:].

Isso cria um outro objeto array com o resultado do filtro, que nesse caso é todos os registros. Veja:

3 - Utilizando o construtor _list()

Neste post estou chamando list do python de array para manter a linguagem comum com outras linguagens de programação, visto que o modo literal com colchetes (exemplo [1,2,3]) cria um objeto do tipo list.

Mas em python um objeto array é um pouco diferente de um objeto list. Vou aprofundar na diferença entre eles em outro post.

Agora que você descobriu que o que vc achava que era um array na verdade é um objeto list (=P), você pode criar um novo objeto utilizando o construtor da classe list passando o objeto atual como parâmetro. Ou seja b = list(a).

Veja o resultado, vou aproveitar o gif pra mostrar o resultado de type([1,2,3]):

4 - Utilizando a biblioteca copy

Por fim, também podemos importar o pacote copy e utilizar a função também de nome copy. Assim:

a = [1,2,3]
import copy
b = copy.copy(a)

O resultado é o mesmo, ou seja, copiamos o objeto list e não só o valor da variável a. Veja:

Copia profunda em array de objetos

Até agora nós falamos de arrays simples de tipos primitivos criando arrays de números inteiros.

Entretanto também podemos criar arrays de objetos, na verdade de endereços de outros objetos ;).

Talvez você já tenha captado o problema em fazer uma cópia simples de um array desse tipo, mas se não pegou o problema eu explico. Dê uma olhada na imagem abaixo.

Quando copiamos um array das 4 formas como falamos acima, nós estamos copiando o objeto array, mas se tivermos um array de objetos (endereços para outros objetos), os objetos apontados de dentro do array não são copiados.

Aí a gente cai no mesmo problema do início, mas em um nível mais baixo.

Veja por exemplo o código abaixo.

class X:
    y = 0
    def __init__(self, y):
        self.y = y

a = [X(y=1), X(y=2), X(y=3)]

Neste ponto, nós criamos três objetos da classe X e um objeto list e apontamos a variável a para o objeto list.

O que acontece se executarmos o código abaixo?

b = a.copy()
b[0].y = 4

Exatamente! Alteramos o valor da variável y do mesmo objeto que é apontado por a[0].

Veja:

Perceba no final do gif que ao imprimir os objetos a e b os valores dos arrays (endereços de memória) são os mesmos. Ou seja, os mesmos objetos, igual a imagem acima.

Se quisermos que os objetos dos arrays também sejam copiados e assim por diante (se for array de arrays de arrays, ...), ou seja uma cópia profunda, temos que usar a função copy.deepcopy(<list>).

Claro que esse método custa mais processamento, mas ao final teremos um array completamente copiado de forma segura. Assim:

import copy
b = copy.deepcopy(a)

Veja o exemplo:

Viu, perceba que o endereço dos objetos internos mudam! Por isso a alteração na variável do objeto b[0].y não afetou a variável do objeto a[0].y, pois a[0] e b[0] agora apontam para objetos diferentes!

Espero que essa explicação tenha expandido um pouco o seu entendimento sobre orientação a objetos.

Esse comportamento é mais ou menos o mesmo em outras linguagens ditas orientadas a objetos como java, javascript, php, etc.

Comente aí em baixo se tiver alguma dúvida, dica, elogio ou sugestões!

Até a próxima!

Referências:

1. Doc: copy
2. Doc: array
3. Doc: list