def _simple_slice(arr, inds, axis):
    """
    This does the same as np.take() except only supports simple slicing, 
    not advanced indexing, and thus is much faster.
    
    References
    ----------
    https://stackoverflow.com/a/37729566
    """
    sl = [slice(None)] * arr.ndim
    sl[axis] = inds
    return arr[tuple(sl)]
