Shuffle 2D numpy array within each row

shufflebyrow.py

def shufflebyrow(X):
    """
    Shuffle each row of a given array independently. Can be useful for
    randomizing binned spikes from all cells, for statistical testing.

    Parameters
    --------

    X:  np.array with two dimensions
        Rows should correspond to cells, columns to time bins.

    Example
    ------

    >>> allspikes.shape
    (37, 10000)
    >>> shuffledspikes = shufflebyrow(allspikes)
    >>> np.allclose(allspikes.sum(axis=1),
                    shuffledspikes.sum(axis=1))
    True

    Notes
    -----
    Adopted from https://github.com/numpy/numpy/issues/5173#issuecomment-467625388
    """
    return np.apply_along_axis(np.random.permutation, axis=1, arr=X)