Split a list randomly given chunk sizes in python

Question

I want to write a function that returns a list of randomly splitted chunks, given a list of sizes of each chunk. For example, say I have a list of unique integers from 0 to 9

lst = list(range(10))

I want to split into 4 chunks of sizes 1, 2, 2 and 5, so the function should take an input variable sizes as below:

my_func(lst, sizes=[1, 2, 2, 5])

I expect the function to return something like

[[3], [1, 7], [2, 4], [0, 5, 6, 8, 9]]

If the input sizes = [5, 1, 4], I expect the function to return something in order like

[[1, 3, 4, 6, 8], [9], [0, 2, 5, 7]]

Any suggestions?

okay. I think I got it after you remind me of random.shuffle. — Shaun Han
– Shaun Han, Commented Mar 5, 2021 at 10:16

Kuldeep Singh Sidhu · Accepted Answer · 2021-03-05 10:18:25Z

2

My approach is to use random.shuffle and then go through the shuffled array!

import random
def my_func(lst, sizes):
    random.shuffle(lst)
    ret = []
    pointer = 0
    for s in sizes:
        ret.append(lst[pointer:pointer+s])
        pointer+=s
    return ret

lst = list(range(10))
print(my_func(lst, sizes = [1, 2, 2, 5]))
print(my_func(lst, sizes = [5, 1, 4]))

[[5], [6, 7], [4, 0], [8, 3, 9, 2, 1]]
[[5, 2, 4, 3, 0], [8], [6, 7, 9, 1]]

edited Mar 5, 2021 at 10:18

answered Mar 5, 2021 at 10:17

Kuldeep Singh Sidhu

3,8762 gold badges16 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Abdul Aziz Barkat · Accepted Answer · 2021-03-05 10:18:11Z

0

Use random.shuffle to shuffle your list. Also just keep slicing the list.

import random

def random_split(lst, sizes):
    if sum(sizes) != len(lst):
        raise ValueError("Sizes should match list length!")
    lst = lst.copy() # Copy list
    random.shuffle(lst) # shuffle
    res = []
    for size in sizes:
        split_list, lst = lst[:size], lst[size:] # Make two pieces of the list
        # split_list.sort()
        # Above line should be used if each sublist should be sorted
        res.append(split_list)
    return res

answered Mar 5, 2021 at 10:18

Abdul Aziz Barkat

22.3k9 gold badges104 silver badges94 bronze badges

2 Comments

Kuldeep Singh Sidhu Over a year ago

Hi mate, this is a duplicate answer!

Abdul Aziz Barkat Over a year ago

@KuldeepSinghSidhu Not exactly both were in short succession. Honest mistake friend. :)

Vivek G. · Accepted Answer · 2021-03-05 10:24:56Z

0

you can do something like this:

import random

lis=[]
g_lis=[]

for i in sizes:
    list_int = i

    for j in range(list_int):
        num = random.randint(0,9)
        lis.append(num)

    g_lis.append(lis)

answered Mar 5, 2021 at 10:24

Vivek G.

611 silver badge7 bronze badges

Collectives™ on Stack Overflow

Split a list randomly given chunk sizes in python

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related