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']

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

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