Search code examples
pythonpartitiongroup

Partition a list items using a list containing partition numbers


Given a list

lst=['a','a','b','b','b','c','d','d']

and a list 'l' containing partition numbers

l=[2,3,1,2]

what I want is

partitioned_lst=[['a','a'],['b','b','b'],['c'],['d','d']]

Solution

  • partitioned_lst=[]
    i=0
    for n in l:
        partitioned_lst.append(lst[i:i+n])
        i+=n
    partitioned_lst