在 Pandas DataFrame 中增加总计行(列)

 def add_total(df, axis=0):

    """
    在 Pandas DataFrame 中增加总计行(列)

    参数:
        df (pandas.DataFrame):需要增加总计行(列)的 DataFrame。
        axis (int, optional): 0 表示对列进行求和,增加总计行;1 表示对行进行求和,增加总计列。默认为0。
    抛出:
        ValueError:如果 axis 参数不为 0 或 1,则报错。

    返回:
        pandas.DataFrame:增加了总计行(列)的新 DataFrame。
    """
    df = df.copy()
    if axis == 0:
        df.loc['Total'] = df.sum(axis=0, numeric_only=True)
    elif axis == 1:
        df['Total'] = df.sum(axis=1, numeric_only=True)
    else:
        raise ValueError('axis parameter should be 0 or 1')
    return df


# 0 增加1行对列进行求和,增加1 列对行进行求和
df = (
    df.pipe(add_total, axis=1)
)

评论

此博客中的热门博文

V2ray websocket(ws)+tls+nginx分流

Rstudio 使用代理