Extraindo o primeiro item de cada sublista em Python

Em Python, ao trabalhar com estruturas de dados aninhadas, como listas de listas, pode ser necessário para extrair elementos específicos de cada sublista para criar uma nova lista. Por exemplo, para extrair o primeiro item de cada sublista em uma lista de listas, você pode usar os seguintes métodos:

1. Compreensão de lista:

Essa abordagem concisa utiliza a compreensão de lista para iterar em cada sublista e anexar o primeiro item a uma nova lista.

lst = [['a', 'b', 'c'], [1, 2, 3], ['x', 'y', 'z']]
lst2 = [item[0] for item in lst]

2. Mapa e Lambda:

Você pode empregar a função map() junto com uma função lambda para obter o mesmo resultado. A função lambda recupera o primeiro item de cada sublista.

lst = [['a', 'b', 'c'], [1, 2, 3], ['x', 'y', 'z']]
lst2 = list(map(lambda sublist: sublist[0], lst))

3. For Loop:

Embora um pouco menos eficiente que os métodos anteriores, um loop for também pode realizar a tarefa.

lst = [['a', 'b', 'c'], [1, 2, 3], ['x', 'y', 'z']]
lst2 = []
for sublist in lst:
    lst2.append(sublist[0])

A escolha apropriada depende dos requisitos específicos e das considerações de desempenho do seu código. Para cenários simples e eficientes, compreensão de lista ou map() com lambda são comumente preferidos.