Como Usar List Comprehensions em Python

• Por: Jonas
• 7 de Outubro de 2024 às 16:46
• Views: 160

Como Usar List Comprehensions em Python

Introdução

Uma das características mais poderosas e elegantes do Python é a lista de compreensão (list comprehension). Essa sintaxe concisa permite criar listas de maneira rápida e eficiente, em apenas uma linha de código, economizando tempo e tornando o código mais legível. Neste post, você vai aprender como criar uma lista de compreensão em Python e entender as situações em que ela é a melhor escolha.

O que é uma Lista de Compreensão?

Em Python, uma lista de compreensão é uma maneira de criar listas de forma concisa, usando menos linhas de código. A sintaxe básica envolve uma expressão seguida de um for e, opcionalmente, um ou mais filtros if. Vamos ver um exemplo básico para ilustrar isso.

Sintaxe de uma Lista de Compreensão

A estrutura de uma lista de compreensão segue este formato:

[expressao for item in iteravel]

Essa sintaxe gera uma nova lista ao iterar sobre um objeto, aplicando uma expressão a cada item no iterável. Se desejar, pode incluir um filtro com if:

[expressao for item in iteravel if condicao]

Exemplo Prático de Lista de Compreensão

Vamos ver um exemplo simples: gerar uma lista com os quadrados dos números de 0 a 9 usando a lista de compreensão:

quadrados = [x**2 for x in range(10)]
print(quadrados)

Saída:

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

Neste exemplo, a expressão x**2 é aplicada a cada número do range(10), gerando a lista de quadrados. A lista resultante contém os números de 0 ao quadrado até 9 ao quadrado.

Usando Condições em Lista de Compreensão

Agora vamos usar uma condição para criar uma lista que só contenha os números pares:

pares = [x for x in range(10) if x % 2 == 0]
print(pares)

Saída:

[0, 2, 4, 6, 8]

Aqui, a condição if x % 2 == 0 garante que apenas números pares sejam incluídos na lista.

Vantagens das Listas de Compreensão

• Sintaxe concisa: Torna o código mais curto e fácil de ler.
• Performance: Geralmente, as listas de compreensão são mais rápidas do que loops tradicionais.
• Facilidade de manutenção: Menos linhas de código significam menos espaço para erros e mais clareza na lógica.

Quando Evitar o Uso de Lista de Compreensão

Embora listas de compreensão sejam muito úteis, elas podem reduzir a legibilidade do código se forem complexas demais. Se sua lista de compreensão tiver múltiplos if e expressões aninhadas, considere usar loops tradicionais para manter o código mais claro.

Exemplo Avançado: Lista de Compreensão Aninhada

Você também pode usar listas de compreensão aninhadas. Veja este exemplo, que gera uma matriz (lista de listas):

matriz = [[x for x in range(3)] for y in range(3)]
print(matriz)

Saída:

[[0, 1, 2], [0, 1, 2], [0, 1, 2]]

Neste exemplo, a lista de compreensão é usada dentro de outra para gerar uma matriz 3x3.

Comparação: List Comprehension vs Loop For Tradicional

quadrados = []
for x in range(10):
    quadrados.append(x**2)
print(quadrados)

quadrados = [x**2 for x in range(10)]
print(quadrados)