python split range into n groups code example

Example 1: python split range equally

def chunks(lst, n):
    """Yield successive n-sized chunks from lst."""
    for i in range(0, len(lst), n):
        yield lst[i:i + n]
        
list(chunks(range(10, 75), 10))

Example 2: python split range into n groups

import numpy
x = range(25)
l = numpy.array_split(numpy.array(x),6)

Example 3: python divide array into n parts

def split(a, n):
    k, m = divmod(len(a), n)
    return (a[i * k + min(i, m):(i + 1) * k + min(i + 1, m)] for i in range(n))
print( list(split(range(11), 3)) )  # [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10]]

Tags:

Misc Example