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

Question

我的问题很简单，我有一个列表：

my_list = [['a1','b1'],['a2','b2'],['a3','b3','c3'],['a4','b4','c4','d4','e4']]

如何轻松获取每个列表的最后一个元素，即：

[ 'b1' , 'b2' , 'c3' , 'e4' ]

谢谢！

Answer 1

您可以使用索引-1获取每个元素的最后一个元素，只需对所有子列表执行此操作。

print [item[-1] for item in my_list]
# ['b1', 'b2', 'c3', 'e4']

如果你在寻找一种习惯的方式，那么你可以

import operator
get_last_item = operator.itemgetter(-1)
print map(get_last_item, my_list)
# ['b1', 'b2', 'c3', 'e4']
print [get_last_item(sub_list) for sub_list in my_list]
# ['b1', 'b2', 'c3', 'e4']

如果您使用的是Python3.x，那么您也可以这样做

print([last for *_, last in my_list])
# ['b1', 'b2', 'c3', 'e4']

Answer 2

你可以试试

这是demo。

>>> [i.pop() for i in my_list]
['b1', 'b2', 'c3', 'e4']

或

>>> my_list = [['a1','b1'],['a2','b2'],['a3','b3','c3'],['a4','b4','c4','d4','e4']]
>>> [i[-1] for i in my_list]
['b1', 'b2', 'c3', 'e4']

Answer 3

使用map的替代方案：

last_items = map(lambda x: x[-1], my_list)

或：

from operator import itemgetter
print map(itemgetter(-1), my_list)