Извлечение первого элемента из каждого подсписка в Python

В Python при работе с вложенными структурами данных, такими как списки списков, это может потребоваться для извлечения определенных элементов из каждого подсписка для создания нового списка. Например, чтобы извлечь первый элемент из каждого подсписка в списке списков, вы можете использовать следующие методы:

1. Понимание списка:

Этот краткий подход использует понимание списка для перебора каждого подсписка и добавления первого элемента в новый список.

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

2. Карта и лямбда:

Вы можете использовать функцию map() вместе с лямбда-функцией для достижения того же результата. Лямбда-функция извлекает первый элемент в каждом подсписке.

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

3. Цикл for:

Хотя цикл for несколько менее эффективен, чем предыдущие методы, он также может выполнить эту задачу.

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

Соответствующий выбор зависит от конкретных требований и показателей производительности вашего кода. Для простых и эффективных сценариев обычно предпочтительнее использовать понимание списка или карту() с лямбда-выражением.