ratio with groupby pandas

Question

I have a dataframe with 3 columns, product_id, user_city, purchase (0 or 1) to see if the customer purchased or not the item. I want to calculate the percentage of purchase by city For example, if my dataframe looks like this:

product_id   user_city  purchase 
    1          NYC         0
    2          NYC         1
    3          LA          1
    4          LA          1
    5          LA          1
    6          LA          0

here I would get:

      perc_purchase
NYC    0.5
LA     0.75

I have used groupedby with size() but this only gives me a count of 0 and 1 by city, no percentage. Please, help!!

`df.groupby('user_city')['purchase'].mean().to_frame('perc_purchase')` — ansev, Dec 29 '19 at 09:27

ignoring_gravity · Answer 1 · 2019-12-29T09:33:58.010

0

You should do

df.groupby('user_city')['purchase'].mean().sort_values().to_frame('purchase')

This will give you

           purchase
user_city          
NYC            0.50
LA             0.75

edited Dec 29 '19 at 09:33

answered Dec 29 '19 at 09:23

ignoring_gravity

6,677
4
32
65

This is great, thank you very much. do you know if there is a way to name this new column? – PandasKoala Dec 29 '19 at 09:32
@ansev replied above to this! – PandasKoala Dec 29 '19 at 09:33
@PandasKoala, yes, just updated my answer to reflect that :) – ignoring_gravity Dec 29 '19 at 09:34
yes! just saw that as well :-) thanks a lot. Last question, let's say I want to display as well a column with the count of 0s and a column with the count of 1s. do you know how to do that? – PandasKoala Dec 29 '19 at 09:37

ratio with groupby pandas

1 Answers1