Múltiplos loops em compreensão de listas em Python
Trabalhar com compreensão permite escrever múltiplos loops para gerar uma nova lista. O primeiro loop especificado será o externo, onde o segundo loop especificado será executado. Isso significa que o segundo loop será aninhado dentro do primeiro.
Vamos formar uma lista de tuplas. Para
isso, primeiro escrevemos duas variáveis i
e j, entre parênteses,
denotando uma tupla. Em seguida, iniciamos um loop
com a variável i, que deve
gerar o primeiro número da tupla no
intervalo de 1 a 3. Depois
escrevemos um loop onde o segundo número j
é gerado de 1 a 2:
lst = [(i, j) for i in range(1, 4) for j in range(1, 3)]
print(lst)
Resultado da execução do código:
[(1, 1), (1, 2), (2, 1), (2, 2), (3, 1), (3, 2)]
Dadas duas listas:
lst1 = ['1', '2']
lst2 = ['a', 'b', 'c']
Usando compreensão, crie uma nova lista a partir delas:
[('1', 'a'), ('1', 'b'), ('1', 'c'), ('2', 'a'), ('2', 'b'), ('2', 'c')]
