mymath.py 文件源码

def mode(x, axis = None, keepdims = False):
    from scipy.stats import mode as sp_mode
    mode_x, _ = sp_mode(x, axis = axis)
    if not keepdims:
        mode_x = np.take(mode_x, 0, axis = axis)
    return mode_x