chunk.py

def chunk(xs, n):
    """ Split an iterable into chunks.

    Parameters
    ----------
    xs : iterable
        Iterable to be split.
    n : int
        Chunk size.

    Returns
    -------
    generator
    """

    sublist = []
    for x in xs:
        sublist.append(x)
        if len(sublist) == n:
            yield sublist
            sublist = []
    else:
        # Yield leftover item(s) when input collection
        # is exhausted, if any exist.
        if sublist:
            yield sublist