python - 如何统计属于每个组的条目数？

Question

这是我的数据框:

df = 

GROUP    GRADE   TOTAL_SERVICE_TIME    TOTAL_WAIT_TIME
AAA      1       45                    20
AAA      4       40                    23
AAA      5       35                    21
BBB      2       30                    24
BBB      3       55                    22

我想按 GROU 和 GRADE 对条目进行分组，估计每个分组的平均 TOTAL_SERVICE_TIME 和平均 TOTAL_WAIT_TIME , 以及计算属于每个组的条目数。

我不知道如何进行计数:

output = df.groupby(['GROUP','GRADE'])
           .agg({'TOTAL_SERVICE_TIME' : 'mean', 'TOTAL_WAIT_TIME' : 'mean'})
           .value_counts()
           .reset_index()

我还尝试添加 , 'COUNT' : 'count'，但列 COUNT 应该已经存在。

Answer 1

你很接近，agg 上的文档很简单:

df.groupby(['GROUP','GRADE']).agg({'TOTAL_SERVICE_TIME' : 'mean',
                                   'TOTAL_WAIT_TIME' : ['mean', 'count']})
Out[45]: 
            TOTAL_WAIT_TIME       TOTAL_SERVICE_TIME
                       mean count               mean
GROUP GRADE                                         
AAA   1                  20     1                 45
      4                  23     1                 40
      5                  21     1                 35
BBB   2                  24     1                 30
      3                  22     1                 55