Создать список списков, где длина каждого элемента определяется в другом списке

Question

Если у меня есть список чисел, как я могу создать список списков, где длина каждого элемента предопределена в другом списке чисел?

Пример:

list = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]
lengths = [6,5,1,3,1] #in ascending order
result = [[1,2,3,4,5,6],[7,8,9,10,11],[12],[13,14,15],[16]]

Есть ли общий c способ сделать это, предполагая, что sum(lengths) = len(list) всегда выполняется?

Answer 1

Попробуйте это:

s=0
l=[]
for i in lengths:
  l.append(list[s:s+i])
  s+=i

Вывод

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