Python Adicionando Itens

A lista de compras nunca fica pronta de uma vez. A Ana abre o armário, vê que o café acabou e escreve "café" no fim da folha; a vizinha pede para ela trazer sal e óleo, e mais dois itens entram. Na aula Alterando Itens, você viu que compras[4] = "sabão" dá erro numa lista de 4 itens: a atribuição só troca o que já existe. Para acrescentar itens, a lista tem métodos próprios. É o assunto desta aula.

A ideia: escrever no fim da folha

Quando a Ana lembra de um item novo, o jeito mais natural é escrevê-lo no fim da lista, embaixo do último. Às vezes, porém, ela quer encaixar um item no meio ("o pão tem de ser o primeiro, para eu não esquecer"). E às vezes recebe uma listinha inteira de outra pessoa para juntar à dela. São três situações, e cada uma tem o seu método:

Os três alteram a própria lista (ela é mutável, como você viu em Alterando Itens). Guarde essa frase: ela é a chave do erro mais famoso desta aula.

1. append(): um item no fim

compras = ["arroz", "feijão"]
print(compras)
compras.append("café")
print(compras)
compras.append("leite")
print(compras)
print(f"Agora são {len(compras)} itens.")

compras

A lista que vai receber o item.

.

O ponto: "da lista compras, use o método…". Métodos são funções que pertencem a um valor.

append

O método que acrescenta um item no fim.

("café")

Entre parênteses, o item novo. Exatamente um.

$ python3 append.py
['arroz', 'feijão']
['arroz', 'feijão', 'café']
['arroz', 'feijão', 'café', 'leite']
Agora são 4 itens.

A cada append, a lista ganhou um item no fim e o len aumentou 1. Repare que a linha compras.append("café") não tem =: ela não cria uma lista nova para guardar em algum lugar. Ela muda a lista compras que já existe.

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

print(compras)

compras = ["arroz", "feijão"]
compras.append("leite")
compras.append("pão")
print(compras)

2. Montando uma lista do zero

O uso mais comum do append é encher uma lista vazia, um item de cada vez. A Ana quer que o programa pergunte os itens e monte a lista para ela:

compras = []
for i in range(3):
    item = input("Item: ")
    compras.append(item)
print(compras)

$ python3 montar.py
Item: ovos
Item: pão
Item: leite
['ovos', 'pão', 'leite']

1. compras = []: a lista começa vazia. Ela precisa existir antes do laço, senão não há onde fazer o append.
2. O for i in range(3) (aula range()) repete o bloco 3 vezes.
3. A cada volta, o input() lê um item e o append o põe no fim.
4. Depois do laço, a lista tem os 3 itens, na ordem em que foram digitados.

E se a Ana não souber quantos itens vão ser? Um while com uma palavra para parar (aulas while e break e continue):

compras = []
while True:
    item = input("Item (ou fim): ")
    if item == "fim":
        break
    compras.append(item)
print(f"Sua lista tem {len(compras)} itens: {compras}")

$ python3 montar2.py
Item (ou fim): café
Item (ou fim): leite
Item (ou fim): fim
Sua lista tem 2 itens: ['café', 'leite']
$ python3 montar2.py
Item (ou fim): fim
Sua lista tem 0 itens: []

Na segunda vez, a Ana digitou fim logo de cara, e a lista ficou vazia, sem erro nenhum. Repare que o "fim" não entrou na lista: o break vem antes do append.

3. insert(): um item numa posição

Na fila da cantina, a Ana chegou primeiro e precisa entrar no começo; depois o Beto vai entrar no meio. O insert recebe dois valores: primeiro o índice onde o item vai ficar, depois o item.

fila = ["Bruno", "Carla", "Davi"]
fila.insert(0, "Ana")
print(fila)
fila.insert(2, "Beto")
print(fila)
fila.insert(100, "Zeca")
print(fila)

$ python3 insert.py
['Ana', 'Bruno', 'Carla', 'Davi']
['Ana', 'Bruno', 'Beto', 'Carla', 'Davi']
['Ana', 'Bruno', 'Beto', 'Carla', 'Davi', 'Zeca']

fila.insert(0, "Ana")

Põe a Ana no índice 0, o começo. Todos os outros andam uma casa para a direita: o Bruno, que era o 0, passa a ser o 1.

fila.insert(2, "Beto")

O Beto fica no índice 2, ou seja, antes de quem estava no 2 (a Carla).

fila.insert(100, "Zeca")

Um índice maior que a lista não dá erro: o item simplesmente vai para o fim, como num append.

Com um índice negativo, cuidado: insert(-1, item) não põe no fim. Ele põe antes do último, porque o item novo ocupa o índice -1 e empurra quem estava lá:

>>> compras = ["arroz", "feijão"]
>>> compras.append("café")
>>> print(compras.append("leite"))
None
>>> compras
['arroz', 'feijão', 'café', 'leite']
>>> compras.insert(-1, "sal")
>>> compras
['arroz', 'feijão', 'café', 'sal', 'leite']

O sal ficou entre o café e o leite. Para pôr no fim, use append. (E repare na terceira linha: print(compras.append("leite")) mostrou None, mas o leite entrou na lista. Guarde isso para o Quando dá errado.)

4. Juntando listas: extend(), + e +=

A vizinha mandou a listinha dela: sal e óleo. A Ana quer juntar os itens dela à lista de compras.

compras = ["arroz", "feijão"]
da_vizinha = ["sal", "óleo"]
compras.extend(da_vizinha)
print(compras)
print(da_vizinha)

$ python3 extend.py
['arroz', 'feijão', 'sal', 'óleo']
['sal', 'óleo']

O extend acrescentou, no fim de compras, cada um dos itens de da_vizinha. A lista da vizinha ficou como estava.

append de uma lista: a lista dentro da lista

E se a Ana usar append com a lista da vizinha?

compras = ["arroz", "feijão"]
compras.append(["sal", "óleo"])
print(compras)
print(len(compras))

$ python3 append_lista.py
['arroz', 'feijão', ['sal', 'óleo']]
3

Repare nos colchetes de dentro: o append põe um item no fim, e esse item foi a lista inteira da vizinha. A lista ficou com 3 itens, e o terceiro é uma lista (você viu listas dentro de listas em Acessando Itens). Às vezes é isso mesmo que você quer; quase sempre, quem escreve isso queria o extend.

O + e o +=

Como nos textos (aula Juntar e Repetir Textos), o + junta duas listas. Mas há uma diferença importante: o + cria uma lista nova e deixa as duas originais em paz. Já o += muda a lista da esquerda, como o extend.

>>> compras = ["arroz", "feijão"]
>>> nova = compras + ["sal", "óleo"]
>>> nova
['arroz', 'feijão', 'sal', 'óleo']
>>> compras
['arroz', 'feijão']
>>> compras += ["pão"]
>>> compras
['arroz', 'feijão', 'pão']

• nova = compras + ["sal", "óleo"]: a lista nova tem os 4 itens; compras continua com 2. O + não mexeu em ninguém.
• compras += ["pão"]: agora sim compras mudou. O += em listas funciona como o extend. Do lado direito, uma lista (por isso ["pão"], com colchetes).

Repetindo: *

Outro operador que vem dos textos: [0] * 3 é uma lista com o 0 repetido 3 vezes. É um jeito rápido de criar o placar de um jogo com três jogadores, todos começando do zero:

>>> placar = [0] * 3
>>> placar
[0, 0, 0]

Formas diferentes de adicionar

Quando dá errado

compras = ["arroz", "feijão"]
compras = compras.append("café")
print(compras)
compras.append("leite")

$ python3 none.py
None
Traceback (most recent call last):
  File "/home/ana/none.py", line 4, in <module>
    compras.append("leite")
    ^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'append'

compras = []
for i in range(3):
    item = input("Item: ")
    compras = compras.append(item)
print(compras)

$ python3 none2.py
Item: ovos
Item: pão
Traceback (most recent call last):
  File "/home/ana/none2.py", line 4, in <module>
    compras = compras.append(item)
              ^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'append'

>>> compras.extend("óleo")
>>> compras
['arroz', 'feijão', 'café', 'sal', 'leite', 'ó', 'l', 'e', 'o']

>>> compras = ["arroz", "feijão", "pão"]
>>> compras += "sal"
>>> compras
['arroz', 'feijão', 'pão', 's', 'a', 'l']

>>> compras = ["arroz", "feijão", "pão"]
>>> compras + "sal"
Traceback (most recent call last):
  File "<python-input-1>", line 1, in <module>
    compras + "sal"
    ~~~~~~~~^~~~~~~
TypeError: can only concatenate list (not "str") to list

>>> compras.append("ovos", "farinha")
Traceback (most recent call last):
  File "<python-input-2>", line 1, in <module>
    compras.append("ovos", "farinha")
    ~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^
TypeError: list.append() takes exactly one argument (2 given)
>>> compras.push("ovos")
Traceback (most recent call last):
  File "<python-input-3>", line 1, in <module>
    compras.push("ovos")
    ^^^^^^^^^^^^
AttributeError: 'list' object has no attribute 'push'

Referência rápida

Exercícios

Missão final: a lista pelo teclado

compras = []
for i in range(3):
    item = input("Item: ")
    compras = compras.append(item)
print(compras)

compras = []
for i in range(3):
    item = input("Item: ")
    compras.append(item)
print(compras)

Resumo

• append(x) põe um item no fim; é o jeito mais usado de encher uma lista que começou com [].
• insert(i, x) põe o item no índice i; os outros andam para a direita. insert(-1, x) põe antes do último.
• extend(outra) e += põem os itens de outra lista no fim; append de uma lista põe a lista inteira como um item só.
• lista + outra cria uma lista nova e não muda nenhuma das duas.
• Os métodos que alteram a lista devolvem None: nunca escreva lista = lista.append(x).
• 'NoneType' object has no attribute 'append' quer dizer que a variável recebeu None antes; procure o = com um método de lista.
• Texto no extend ou no += vira letras soltas.