python - 使用 groupby 获取总行数和值 > 2 的行数

Question

如何为以下数据框计算值 > 2 的行的大小和数量:

df
     kpi_date    cell_name         call_drop
    2016-01-01    bgl_2345            0.2
    2016-01-01    bgl_2346            2.3
    2016-01-01    blg_2347            0.3
    2016-01-02    bgl_2345            1.4
    2016-01-02    bgl_2346            2.5
    2016-01-03    bgl_2347            2.7

输出应该是

  kpi_date    call_drop   call_Drop>2
   2016-01-01    3             1
   2016-01-02    2             1
   2016-01-03    1             1

请指导

Answer 1

循序渐进:

grouped = df.groupby('kpi_date')
count = grouped['call_drop'].count()
greater2 = grouped.apply(lambda x: x['call_drop'][x['call_drop']> 2].count())
greater2.name = 'call_Drop>2'
print(pd.concat([count, greater2], axis=1))

输出:

            call_drop  call_Drop>2
kpi_date                          
2016-01-01          3            1
2016-01-02          2            1
2016-01-03          1            1