how can i sum dataframe columns on other columns in python?

Question

I have the following data frame.

ID    Product  quantity
9626     a       1
9626     b       1
9626     c       1
6600     f       1
6600     a       1
6600     d       1

And I want to join rows by ID. Below is an example of the results. (The quantity column is optional. This column is not necessary.)

ID    Product  quantity
9626     a,b,c     3
6600     a,d,f     3

I used merge and sum, but it did not work.

Is this problem solved only with a loop statement?

I'd appreciate it if you could provide me with a solution.

cs95 · Accepted Answer · 2019-01-23 05:35:41Z

5

Use groupby.agg:

df = (df.sort_values('Product')
        .groupby('ID', as_index=False, sort=False)
        .agg({'Product':','.join, 'quantity':'sum'}))

print(df)
     ID Product  quantity
0  9626   a,b,c         3
1  6600   a,d,f         3

edited Jan 23, 2019 at 5:35

cs95

406k106 gold badges744 silver badges797 bronze badges

answered Jan 23, 2019 at 4:56

Space Impact

13.3k26 silver badges51 bronze badges

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

Collectives™ on Stack Overflow

how can i sum dataframe columns on other columns in python?

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