Prática: transformar dados de alunos

Os dados de partida

Vamos trabalhar com um formato que você vai encontrar sempre: uma lista de dicionários, em que cada dicionário é um aluno com nome e uma lista de notas. É assim que dados costumam chegar de um arquivo JSON ou de uma planilha convertida. O objetivo é responder três perguntas com compreensões: quem foi aprovado, quais são todos os nomes e qual a média de cada aluno. Comece copiando os dados abaixo no Playground.

alunos = [
    {"nome": "Ana", "notas": [8, 9, 10]},
    {"nome": "Bruno", "notas": [4, 5, 6]},
    {"nome": "Carla", "notas": [7, 7, 8]},
    {"nome": "Diego", "notas": [3, 4, 5]},
]

print(len(alunos), "alunos carregados")  # 4 alunos carregados

A lista de dicionários de partida: cada aluno tem um nome e uma lista de notas.

Antes de filtrar, precisamos da média de cada aluno. A média é a soma das notas dividida pela quantidade, e você já conhece sum e len do Básico. Para não repetir essa conta o tempo todo, vamos primeiro montar uma versão dos dados com a média já calculada. Uma compreensão de lista, percorrendo os alunos, cria uma nova lista de dicionários, cada um com nome e média. Repare que estamos construindo dicionários novos dentro da compreensão.

com_media = [
    {"nome": a["nome"], "media": sum(a["notas"]) / len(a["notas"])}
    for a in alunos
]

for a in com_media:
    print(a["nome"], a["media"])
# Ana 9.0
# Bruno 5.0
# Carla 7.333333333333333
# Diego 4.0

Uma compreensão que monta dicionários novos, cada um com o nome e a média calculada.

Filtrar os aprovados e mapear os nomes

Com as médias prontas, filtrar os aprovados é uma compreensão de lista com if no fim. Digamos que a nota de corte seja seis. A compreensão percorre com_media e mantém só os dicionários cuja média chega a seis. O resultado é uma lista menor, com os mesmos dicionários, só que apenas os dos aprovados. É o filtro da aula 2 aplicado a dados de verdade, e a leitura continua direta: mantenha o aluno, para cada aluno, se a média dele for suficiente.

aprovados = [a for a in com_media if a["media"] >= 6]

for a in aprovados:
    print(a["nome"], a["media"])
# Ana 9.0
# Carla 7.333333333333333

Filtro com if no fim: só entram os alunos com média a partir de seis.

Mapear os nomes é ainda mais curto. Mapear significa transformar cada item em outro, aqui trocando o dicionário inteiro pelo valor de uma chave. A expressão vira a['nome'], que puxa o nome de cada dicionário. Dá para fazer isso com todos os alunos ou só com os aprovados, dependendo do que você precisa. Os dois exemplos abaixo mostram a diferença, e ambos são a mesma técnica de mapeamento vista na aula 1, agora extraindo um campo.

todos_os_nomes = [a["nome"] for a in alunos]
nomes_aprovados = [a["nome"] for a in aprovados]

print(todos_os_nomes)    # ['Ana', 'Bruno', 'Carla', 'Diego']
print(nomes_aprovados)   # ['Ana', 'Carla']

Mapear extraindo um campo: a expressão a["nome"] troca o dicionário pelo nome.

Montar o dicionário de nome para média

A última tarefa junta o filtro com a compreensão de dicionário. Queremos um dicionário que ligue o nome de cada aprovado à sua média, para consultar rápido depois. A forma é {chave: valor for ...}: a chave é o nome, o valor é a média, e um if no fim mantém só os aprovados. Em uma expressão, saem o filtro, o mapeamento das chaves e dos valores, e o dicionário pronto para consulta por nome. É o encontro das aulas 2 e 4 em uma linha só.

nome_media = {
    a["nome"]: round(a["media"], 1)
    for a in com_media
    if a["media"] >= 6
}

print(nome_media)          # {'Ana': 9.0, 'Carla': 7.3}
print(nome_media["Ana"])   # 9.0

Compreensão de dicionário com filtro: nome para média, só dos aprovados, arredondada.

Vale olhar o conjunto do que você fez. Partindo de uma lista de dicionários crua, você calculou médias, filtrou aprovados, mapeou nomes e montou um índice de nome para média, tudo com compreensões, sem um único laço com append. Compare mentalmente com quantas linhas isso levaria no estilo do Básico: criar listas vazias, abrir laços, indentar, dar append. As compreensões não deixaram só mais curto; deixaram cada passo com uma intenção legível. Esse é o ganho concreto do módulo.