在列表

时间:2018-04-25 13:02:49

标签: python

我有一个列表,如:

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]

有什么想法吗?

1 个答案:

答案 0 :(得分:8)

使用列表推导将相同的切片应用于每个子列表:

b = [l[:2] for l in a]