Pandas : Group by and count based on specific value

Question

My deliveries dataframe looks like this :

Each time there is a 0, it means the delivery was made at the same place than the previous one. I would like to create a new column "Number of deliveries at this place" that would look like:

df2

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

I don't figure out how I could do this group by and count by taking into account the last not null (0) value. Many thanks for your help !

Stef · Accepted Answer · 2022-08-22 11:20:07Z

5

You can group by the cumsum of non-zero distances and then count

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

Result:

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

edited Aug 22, 2022 at 11:20

answered Aug 22, 2022 at 11:14

Stef

30.9k3 gold badges34 silver badges60 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Pandas : Group by and count based on specific value

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related