split() for general iterables

split.py

import inspect

def split(iterable, by):
    """Generalized split function that works on any iterable."""
    separator = by if inspect.isfunction(by) else lambda x: x == by
    res = []

    curr = []
    for elem in iterable:
        if separator(elem):
            res.append(curr)
            curr = None
        else:
            curr = curr or []
            curr.append(elem)
    res.append([] if curr is None else curr)

    return res