python: split list into fixed number of evenly distributed sublists

array_split.py

"""
Googling a solution for this seems to always end with [numpy array_split](https://numpy.org/doc/stable/reference/generated/numpy.array_split.html)
This avoids having numpy as dependency just for this
"""

def array_split(array, parts):
    """
    split an array into parts arrays, evenly size

    >>> list(array_split(list(range(12)), 10))
    [[0, 1], [2, 3], [4], [5], [6], [7], [8], [9], [10], [11]]
    >>> list(array_split(list(range(8)), 10))
    [[0], [1], [2], [3], [4], [5], [6], [7]]
    >>> list(array_split(list(range(10)), 10))
    [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9]]
    >>> list(array_split([], 10))
    []
    """
    n = len(array)
    np, nl = divmod(n, parts)
    i = 0
    for p in range(parts if np > 0 else nl):
        s = np + 1 if p < nl else np
        yield array[i:i+s]
        i += s

if __name__ == "__main__":
    import doctest
    doctest.testmod()