python - Pandas : Group by and count based on specific value

Question

我的交货数据框如下所示:

df

     Distance    
1      10
2       0
3       0
4       3
5       2
6       0

每出现一个0，就表示是在与前一次相同的地点交货的。我想创建一个新列“此地点的交货数量”，如下所示:

df2

     Distance    Nb_Deliveries  
1      10              3
2       0              3
3       0              3
4       3              1
5       2              2 
6       0              2

我不知道如何通过考虑最后一个非空(0)值来进行分组和计数。非常感谢您的帮助!

Answer 1

您可以按 cumsum 进行分组非零距离，然后计数

df['Nb_Deliveries'] = df.groupby(df.Distance.ne(0).cumsum()).Distance.transform('count')

结果:

   Distance  Nb_Deliveries
1        10              3
2         0              3
3         0              3
4         3              1
5         2              2
6         0              2