将列表A的第一个元素转换为列表A的前两个元素的列表

Question

在下面的代码中，我有一个简单的整数列表。我希望列表的第一个元素是列表中两个第一个元素的列表。

import numpy as np

distances=[]
for i in range(5):
    distances.append([])
    for j in range(5):
        distances[i].append(np.random.randint(5))
print distances

所以如果我的列表是：[1,2,3,4,5]我希望它是[[1,2]，2,3,4,5]。

Answer 1

代码 -

In [1]: ls = [1,2,3,4,5]

In [2]: ls[0] = ls[:2]

In [3]: ls
Out[3]: [[1, 2], 2, 3, 4, 5]

Answer 2

另一种选择，

newlist = [[a,b],*rest] = ls