我有一个列表,如:
a = [[1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15], [16,17,18,18,20]]
我想创建一个新列表,它只包含一个行代码中每个子列表的前两个元素。例如:
b = [[1, 2], [6, 7], [11, 12], [16, 17]]
我尝试了以下但是它不起作用:
b = a[:][0:2]
有什么想法吗?
答案 0 :(得分:8)
使用列表推导将相同的切片应用于每个子列表:
b = [l[:2] for l in a]