在 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)
)
评论
发表评论