Python: 改变列表列表的形状

Question

使用 python，我想转换此列表列表：

list_3_5 = [[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15]]

一个不同的形状 -- 一个包含 6 个列表的列表，其中前 3 个新列表具有 list_3_5 中 3 个主列表中每个列表的前三个值，然后是最后 3 个新列表剩余的值来自 list_3_5:

[[1,2,3],[6,7,8],[11,12,13],[4,5],[9,10],[14,15]]

感谢任何帮助。

更新：

对于更长的列表，这：

list2 = [list(range(1, 34)), list(range(34, 67)), list(range(67, 100))]

list2b = [list2[i][:5] for i in range(0, len(list2))] + [
    list2[i][5:] for i in range(0, len(list2))
]

或这个

list2 = [list(range(1, 34)), list(range(34, 67)), list(range(67, 100))]
list2b = list(range(2 * len(list2)))

for key, element in enumerate(list2):
    list2b[key] = element[:5]
    list2b[key + len(list2)] = element[5:]

给出这个（修改我原来的 3 个列表列表，每个列表有 33 个元素，变成 6 个列表列表，前三个列表每个有 5 个元素，最后三个列表每个有 28 个）：

[[1, 2, 3, 4, 5],
 [34, 35, 36, 37, 38],
 [67, 68, 69, 70, 71],
 [6, 7, 8, ..., 33], 
 [39, 40, 41, ..., 66],
 [72, 73, 74, ... 99]]

但是我想要的是这个（对于这个例子，一个包含 19 个列表的列表，每个列表有 5 个元素，最后有 1 个列表还有剩余的 4 个元素）：

[[1, 2, 3, 4, 5], 
 [6, 7, 8, 9, 10],
 [11, 12, 13, 14, 15],
 [16, 17, 18, 19, 20],
 ....
 ....
 [91, 92, 93, 94, 95],
 [96, 97, 98, 99]]

看起来像这样，但我不太明白如何针对我的情况进行修改：reshape an irregular list in python

Answer 1

试试这个：

[list_3_5[0][:3],list_3_5[1][:3], list_3_5[2][:3], list_3_5[0][3:],list_3_5[1][3:], list_3_5[2][3:]]

输出：

[[1, 2, 3], [6, 7, 8], [11, 12, 13], [4, 5], [9, 10], [14, 15]]

Answer 2

这里有一个双 for 循环的解决方案，适用于任意长度的列表：

list_3_5 = [[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15]]
newlist = []

for element in list_3_5:
    newlist.append(element[:3])

for element in list_3_5:
    newlist.append(element[3:])    

newlist

Out: [[1, 2, 3], [6, 7, 8], [11, 12, 13], [4, 5], [9, 10], [14, 15]]

但是等等！还有：

list_3_5 = [[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15]]
newlist = list(range(2*len(list_3_5)))

for key, element in enumerate(list_3_5):
    newlist[key] = element[:3]
    newlist[key+len(list_3_5)] = element[3:]

newlist
[[1, 2, 3], [6, 7, 8], [11, 12, 13], [4, 5], [9, 10], [14, 15]]

以下部分回答了问题的更新部分：

biglist = []
for i in list2:
    biglist = biglist + i
#Here we put all small lists in the given list into a big list
print(biglist)

#construct an empty list to save the result as we go:
newlist = []

#add sections of 5 to newlist
while len(biglist) >= 5:
    newlist.append(biglist[:5])
    biglist = biglist[5:]

#add the remaining part of biglist to the newlist:
newlist.append(biglist)
print(newlist)

输出为：

[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15], [16, 17, 18, 19, 20], [21, 22, 23, 24, 25], [26, 27, 28, 29, 30], [31, 32, 33, 34, 35], [36, 37, 38, 39, 40], [41, 42, 43, 44, 45], [46, 47, 48, 49, 50], [51, 52, 53, 54, 55], [56, 57, 58, 59, 60], [61, 62, 63, 64, 65], [66, 67, 68, 69, 70], [71, 72, 73, 74, 75], [76, 77, 78, 79, 80], [81, 82, 83, 84, 85], [86, 87, 88, 89, 90], [91, 92, 93, 94, 95], [96, 97, 98, 99]]

Answer 3

如果你想要一个通用代码来分隔前三个元素，试试这个：

new_list=[list_3_5[i][:3] for i in range(0,len(list_3_5))]+[ list_3_5[i][3:] for i in range(0,len(list_3_5))]

Python: 改变列表列表的形状

Python: changing shape of list of lists

python

list-comprehension

multidimensional-array