在dataframe中，如何按单列分组，并将总和应用于多列，以及添加新的总和列？

Question

这应该是一个简单的问题，但不知何故，我找不到一个有效的解决方案。

我有一个熊猫数据框，看起来像这样：

Slno   Date         col2   col3   col4   col5  col6
0     01/02/20      2      1      2      5     d
1     03/02/20      5      1      2      4     g
2     04/02/20      5      1      2      5     h
3     05/02/20      4      1      2      6     e
4     08/02/20      8      1      2      5     g
5     05/02/20      8      1      2      8     r

**我想按日期分组，并获得col2、col3、col4、col5行的sum（），作为新的列总数

以下是我尝试过的：

df_new[Total] = df.groupby(['Date', sort=False])["col2", "col3", col4", "col5].sum(axis = 1)

它给出ValueError：传递的项目数错误4，位置意味着1

另外我也尝试过

         df = (df.groupby(['Date'])
         .agg(Total=('ConfirmedIndianNational', 'ConfirmedForeignNational', 'Cured', 'Deaths', 'sum'))
         .reset_index())

它提供TypeError:aggregate（）缺少1个必需的位置参数：“arg”

我是新的python，搜索了所有可能的解决方案，但新的用途。

Answer 1

你为什么不做一个简单的：

df[Total] = df.groupby(['Date').col1.sum()+df.groupby(['Date').col2.sum()+df.groupby(['Date').col3.sum()+df.groupby(['Date').col4.sum()

应该没问题。

Answer 2

不清楚的是，每个日期有多个值。如果是这种情况，您的group by应该根据您想要的函数来聚合该列，并使用聚合来执行行的总和。

例如，您需要每个日期每列的最大值：

max_df = df.groupby('Date')(['Date'])["col2", "col3", "col4", "col5"].max()

然后：

max_df.loc[:,'sum_cols'] = max_df[["col2", "col3", "col4", "col5"]].sum(axis = 1)

如果每个日期只有一行，则可以执行以下操作：

df.loc[:,'sum_cols'] = df[["col2", "col3", "col4", "col5"]].sum(axis = 1)

Answer 3

您可以设置Date为索引，然后取轴=1上的列的总和，然后Groupbylevel=0并转换和

df['Total'] = (df.set_index('Date')[["col2", "col3","col4", "col5"]].sum(1)
           .groupby(level=0).transform('sum').to_numpy())

print(df)

   Slno      Date  col2  col3  col4  col5 col6  Total
0     0  01/02/20     2     1     2     5    d     10
1     1  03/02/20     5     1     2     4    g     12
2     2  04/02/20     5     1     2     5    h     13
3     3  05/02/20     4     1     2     6    e     32 # this is duplicated per group
4     4  08/02/20     8     1     2     5    g     16
5     5  05/02/20     8     1     2     8    r     32 # this is duplicated per group