<分区>
我在 python 中有一个列表，我想从中获取一组索引并将其保存为原始列表的子集:
templist = [[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]]
我想要这个:
sublist=[[1, 4, 7, 16, 19,20]]
举个例子。
我无法提前知道列表元素的内容。我所拥有的只是始终相同的索引。
是否有单行方式来执行此操作？
<分区>
我在 python 中有一个列表，我想从中获取一组索引并将其保存为原始列表的子集:
templist = [[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]]
我想要这个:
sublist=[[1, 4, 7, 16, 19,20]]
举个例子。
我无法提前知道列表元素的内容。我所拥有的只是始终相同的索引。
是否有单行方式来执行此操作？
最佳答案
使用 operator.itemgetter :
>>> templist = [[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]]
>>> import operator
>>> f = operator.itemgetter(0,3,6,15,18,19)
>>> sublist = [list(f(templist[0]))]
>>> sublist
[[1, 4, 7, 16, 19, 20]]
关于Python按索引选择列表中的元素，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/23763591/