Рубрики

Python | Получить первый элемент каждого подсписка

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

Примеры:

Input : [[1, 2], [3, 4, 5], [6, 7, 8, 9]]
Output : [1, 3, 6]

Input : [['x', 'y', 'z'], ['m'], ['a', 'b'], ['u', 'v']]
Output : ['x', 'm', 'a', 'u']

Подход № 1: Понимание списка

# Python3 программа для извлечения первого и последнего
# элемент каждого подсписка в списке списков

  

def Extract(lst):

    return [item[0] for item in lst]

      
# Код драйвера

lst = [[1, 2], [3, 4, 5], [6, 7, 8, 9]]

print(Extract(lst))

Выход:

[1, 3, 6]

Подход № 2. Использование zip и распаковки (*)

Этот метод использует zip с * или оператором распаковки, который передает все элементы внутри lst в качестве аргументов функции zip. Таким образом, весь первый элемент станет первым кортежем в сжатом списке. Таким образом, возвращение 0- го элемента решит цель.

# Python3 программа для извлечения первого и последнего
# элемент каждого подсписка в списке списков

  

def Extract(lst):

    return list(list(zip(*lst))[0])

      
# Код драйвера

lst = [[1, 2], [3, 4, 5], [6, 7, 8, 9]]

print(Extract(lst))

Выход:

[1, 3, 6]

Другой способ использования почтового индекса дан ниже:

def Extract(lst):

    return list(next(zip(*lst)))

Подход № 3. Использование itemgetter()

# Python3 программа для извлечения первого и последнего
# элемент каждого подсписка в списке списков

from operator import itemgetter

  

def Extract(lst):

    return list( map(itemgetter(0), lst ))

      
# Код драйвера

lst = [[1, 2], [3, 4, 5], [6, 7, 8, 9]]

print(Extract(lst))

Выход:

[1, 3, 6]

Рекомендуемые посты:

Python | Получить первый элемент каждого подсписка

0.00 (0%) 0 votes