Python Removendo Itens

No mercado, a Ana vai pondo os produtos no carrinho e riscando cada um da lista. Na cantina da escola, quem é atendido sai da fila. No fim do mês, a lista inteira vai para o lixo. Tirar itens de uma lista é tão comum quanto pôr, e o Python tem quatro jeitos de fazer isso, cada um para uma situação: remove(), pop(), del e clear().

A ideia: riscar pelo nome ou pela posição

Há duas maneiras de dizer a alguém qual item riscar da lista:

• Pelo nome: "risca o café". Você não sabe (nem precisa saber) em que posição ele está. É o remove("café").
• Pela posição: "risca o primeiro", "risca o último". Você não liga para qual é o item. É o pop(0), o pop() e o del lista[0].

Em qualquer dos casos, os itens que vêm depois andam uma casa para a esquerda para ocupar o lugar vazio. A lista nunca fica com um "buraco".

1. remove(): tirar pelo valor

compras = ["arroz", "feijão", "café", "leite"]
compras.remove("café")
print(compras)
cesta = ["pão", "leite", "pão", "ovos", "pão"]
cesta.remove("pão")
print(cesta)

$ python3 remove.py
['arroz', 'feijão', 'leite']
['leite', 'pão', 'ovos', 'pão']

compras.remove("café")

"Lista de compras, remova o café". O Python procura o primeiro item igual a "café" e o tira. O leite, que estava no índice 3, passa para o 2.

cesta.remove("pão")

Com itens repetidos, o remove tira só o primeiro que encontrar. Sobraram dois pães.

Como o append (aula Adicionando Itens), o remove muda a própria lista e devolve None. Por isso a linha é só compras.remove("café"), sem =.

Tirando sem dar erro, e tirando todos

Se o item não estiver na lista, o remove para o programa com um erro (você vai vê-lo em Quando dá errado). O jeito seguro é perguntar antes com in. E, para tirar todas as ocorrências, repita enquanto o item ainda estiver lá:

compras = ["arroz", "feijão", "café", "leite"]
item = "açúcar"
if item in compras:
    compras.remove(item)
    print(f"{item} riscado.")
else:
    print(f"{item} não estava na lista.")
cesta = ["pão", "leite", "pão", "ovos", "pão"]
while "pão" in cesta:
    cesta.remove("pão")
print(cesta)

$ python3 remove_seguro.py
açúcar não estava na lista.
['leite', 'ovos']

O while "pão" in cesta: (aula while) remove um pão por volta e para quando não sobra nenhum. Funciona bem em listas pequenas como esta; para listas grandes, o jeito mais rápido é criar uma lista nova só com o que fica, como você vai ver na seção 5.

compras = ["arroz", "feijão", "sabão", "leite"]

print(compras)

compras = ["arroz", "feijão", "sabão", "leite"]
compras.remove("sabão")
print(compras)

2. pop(): tirar pela posição (e ficar com o item)

O pop ("estourar", "tirar de uma vez") tem duas diferenças em relação ao remove: ele recebe o índice, e não o valor; e ele devolve o item que tirou, para você usar. Sem índice, ele tira o último.

>>> compras = ["arroz", "feijão", "café", "leite"]
>>> compras.pop()
'leite'
>>> compras
['arroz', 'feijão', 'café']
>>> compras.pop(1)
'feijão'
>>> compras
['arroz', 'café']

No modo interativo, o 'leite' que aparece logo depois de compras.pop() é o valor devolvido: o item que saiu. Depois, pop(1) tirou o item de índice 1, o feijão.

A fila da cantina

Uma fila é o exemplo perfeito: quem chegou primeiro é atendido primeiro (sai do começo, pop(0)), e o programa precisa saber quem está sendo atendido. É para isso que serve o valor devolvido:

fila = ["Ana", "Bruno", "Carla", "Davi"]
ultimo = fila.pop()
print(f"{ultimo} desistiu e saiu da fila.")
atendido = fila.pop(0)
print(f"Atendendo: {atendido}")
print(f"Ainda na fila: {fila}")

$ python3 fila.py
Davi desistiu e saiu da fila.
Atendendo: Ana
Ainda na fila: ['Bruno', 'Carla']

ultimo = fila.pop()

Tira o último (o Davi) e guarda o nome dele em ultimo. Aqui o = está certo: o pop, ao contrário do append e do remove, devolve algo útil.

atendido = fila.pop(0)

Tira o primeiro (a Ana). O Bruno passa a ser o índice 0.

fila = ["Ana", "Bruno", "Carla"]
atendido = ""
print(f"Atendendo: {atendido}")
atendido = ""
print(f"Atendendo: {atendido}")
print(f"Ainda na fila: {fila}")

fila = ["Ana", "Bruno", "Carla"]
atendido = fila.pop(0)
print(f"Atendendo: {atendido}")
atendido = fila.pop(0)
print(f"Atendendo: {atendido}")
print(f"Ainda na fila: {fila}")

3. del: apagar pelo índice ou uma fatia

del (de delete, "apagar") não é um método: é uma palavra-chave do Python, escrita antes do que você quer apagar, sem ponto e sem parênteses. Ela apaga um item pelo índice, como o pop, mas não devolve nada. E apaga fatias inteiras:

compras = ["arroz", "feijão", "café", "leite", "sabão", "sal"]
del compras[0]
print(compras)
del compras[1:3]
print(compras)
del compras[-1]
print(compras)
compras.clear()
print(compras)

$ python3 del.py
['feijão', 'café', 'leite', 'sabão', 'sal']
['feijão', 'sabão', 'sal']
['feijão', 'sabão']
[]

del compras[0]

Apaga o item de índice 0, o arroz.

del compras[1:3]

Apaga a fatia dos índices 1 e 2 (café e leite), como você viu fatias em Acessando Itens.

del compras[-1]

Apaga o último, o sal.

compras.clear()

O método clear ("limpar") tira todos os itens. A lista continua existindo, só que vazia: [].

del sem índice apaga a variável

Cuidado com a diferença entre esvaziar a lista e apagar a variável:

>>> compras = ["arroz", "café"]
>>> del compras
>>> compras
Traceback (most recent call last):
  File "<python-input-2>", line 1, in <module>
    compras
NameError: name 'compras' is not defined

del compras, sem colchetes, apaga o nome compras: depois disso, é como se a variável nunca tivesse existido (NameError). Para ter uma lista vazia e continuar usando, é compras.clear().

4. Qual usar?

5. Removendo dentro de um for: o item pulado

A professora quer tirar da lista todas as notas abaixo de 6. Parece simples: passar pelas notas e remover as baixas.

notas = [4.0, 5.5, 8.0, 3.0, 9.0, 7.5]
for nota in notas:
    if nota < 6:
        notas.remove(nota)
print(notas)

$ python3 notas_errado.py
[5.5, 8.0, 9.0, 7.5]

O 5.5 escapou, e nenhum erro apareceu. O que aconteceu? O for anda pela lista pelo índice: primeiro o 0, depois o 1, o 2…

1. Volta 1, índice 0: a nota é 4.0. É menor que 6, então sai. A lista vira [5.5, 8.0, 3.0, 9.0, 7.5]: todo mundo andou uma casa para a esquerda, e o 5.5 agora está no índice 0.
2. Volta 2, índice 1: o for segue em frente, para o índice 1, onde agora está o 8.0. O 5.5, que foi para o índice 0, nunca é olhado.
3. Volta 3, índice 2: o 3.0. Sai. O 9.0 e o 7.5 andam para a esquerda.
4. Volta 4, índice 3: o 7.5 (o 9.0 foi para o índice 2 e também foi pulado, mas ele não precisava sair).

É como riscar a lista de compras enquanto uma pessoa a reescreve sem os itens riscados: você perde o lugar. Dois jeitos certos:

notas = [4.0, 5.5, 8.0, 3.0, 9.0, 7.5]
for nota in notas[:]:
    if nota < 6:
        notas.remove(nota)
print(notas)

$ python3 notas_certo.py
[8.0, 9.0, 7.5]

O notas[:] é uma cópia da lista (uma fatia do começo ao fim, aula Copiando Listas). O for anda pela cópia, que não muda, e o remove tira os itens da original. Ninguém perde o lugar.

notas = [4.0, 5.5, 8.0, 3.0, 9.0, 7.5]
aprovadas = []
for nota in notas:
    if nota >= 6:
        aprovadas.append(nota)
print(aprovadas)

$ python3 notas_certo2.py
[8.0, 9.0, 7.5]

O segundo jeito pensa ao contrário: em vez de tirar as notas baixas, ele monta uma lista nova só com as que ficam. A lista original não muda durante o for, então nada é pulado. Este é o jeito mais usado (e o mais rápido em listas grandes), e na aula List Comprehension você vai escrevê-lo numa linha só.

Quando dá errado

compras = ["arroz", "feijão", "café", "leite"]
compras.remove("açúcar")
print(compras)

$ python3 remove_erro.py
Traceback (most recent call last):
  File "/home/ana/remove_erro.py", line 2, in <module>
    compras.remove("açúcar")
    ~~~~~~~~~~~~~~^^^^^^^^^^
ValueError: list.remove(x): x not in list

>>> compras = ["arroz", "café"]
>>> compras.remove(0)
Traceback (most recent call last):
  File "<python-input-1>", line 1, in <module>
    compras.remove(0)
    ~~~~~~~~~~~~~~^^^
ValueError: list.remove(x): x not in list

fila = ["Ana"]
print(fila.pop(0))
print(fila.pop(0))

$ python3 fila_vazia.py
Ana
Traceback (most recent call last):
  File "/home/ana/fila_vazia.py", line 3, in <module>
    print(fila.pop(0))
          ~~~~~~~~^^^
IndexError: pop from empty list

>>> compras = ["arroz", "café"]
>>> compras.pop(10)
Traceback (most recent call last):
  File "<python-input-1>", line 1, in <module>
    compras.pop(10)
    ~~~~~~~~~~~^^^^
IndexError: pop index out of range

>>> compras = ["arroz", "café"]
>>> item = compras.remove("arroz")
>>> print(item)
None

Referência rápida

Exercícios

Missão final: só as notas de aprovação

notas = [4.0, 5.5, 8.0, 3.0, 9.0, 7.5]
for nota in notas:
    if nota < 6:
        notas.remove(nota)
print(notas)

notas = [4.0, 5.5, 8.0, 3.0, 9.0, 7.5]
for nota in notas[:]:
    if nota < 6:
        notas.remove(nota)
print(notas)

Resumo

• remove(x) tira pelo valor, só a primeira ocorrência; se não houver, ValueError. Pergunte antes com in.
• pop(i) tira pelo índice e devolve o item; pop() tira o último. Lista vazia: IndexError.
• del lista[i] e del lista[a:b] apagam pelo índice ou pela fatia; del lista apaga a variável.
• clear() esvazia a lista, que continua existindo.
• Os itens seguintes sempre andam para a esquerda: não fica buraco.
• Para tirar todas as ocorrências: while x in lista: lista.remove(x), ou uma lista nova só com o que fica.
• Nunca remova itens dentro de um for que anda pela mesma lista: ande por uma cópia (lista[:]) ou monte uma lista nova.