📌  相关文章
📜  Python|获取每个子列表的最后一个元素

📅  最后修改于: 2022-05-13 01:55:34.157000             🧑  作者: Mango

Python|获取每个子列表的最后一个元素

给定一个列表列表,编写一个Python程序来提取给定列表列表中每个子列表的最后一个元素。

例子:

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

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


方法#1:列表理解

# Python3 program to extract first and last 
# element of each sublist in a list of lists
  
def Extract(lst):
    return [item[-1] for item in lst]
      
# Driver code
lst = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]
print(Extract(lst))
输出:
[3, 5, 9]


方法 #2:使用zip和 unpacking(*)运算符

此方法使用带有 * 的zip或解包运算符,它将“lst”中的所有项目作为参数传递给 zip函数。提取列表的最后一项有一个小技巧,而不是使用直接压缩,而是使用反向列表迭代器。

# Python3 program to extract first and last 
# element of each sublist in a list of lists
  
def Extract(lst):
    return list(zip(*[reversed(el) for el in lst]))[0]
      
# Driver code
lst = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]
print(Extract(lst))
输出:
(3, 5, 9)

使用zip的另一种方法是将其与Python map一起使用,将reversed作为函数传递。

def Extract(lst):
    return list(list(zip(*map(reversed, lst)))[0])


方法 #3:使用itemgetter()

# Python3 program to extract first and last 
# element of each sublist in a list of lists
from operator import itemgetter
  
def Extract(lst):
    return list( map(itemgetter(-1), lst ))
      
# Driver code
lst = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]
print(Extract(lst))
输出:
[3, 5, 9]