Sum based on grouping in pandas dataframe?

Question

I have a pandas dataframe df which contains:

major       men        women        rank

Art         5          4            1
Art         3          5            3
Art         2          4            2
Engineer    7          8            3
Engineer    7          4            4
Business    5          5            4
Business    3          4            2

Basically I am needing to find the total number of students including both men and women as one per major regardless of the rank column. So for Art for example, the total should be all men + women totaling 23, Engineer 26, Business 17.

I have tried

df.groupby(['major_category']).sum()

But this separately sums the men and women rather than combining their totals.

yatu · Accepted Answer · 2019-07-03 13:47:19Z

2

Just add both columns and then groupby:

(df.men+df.women).groupby(df.major).sum()

major
Art         23
Business    17
Engineer    26
dtype: int64

edited Jul 3, 2019 at 13:47

answered Jul 3, 2019 at 13:46

yatu

88.6k12 gold badges93 silver badges148 bronze badges

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

Comments

anky · Accepted Answer · 2019-07-03 13:45:31Z

2

melt() then groupby():

df.drop('rank',1).melt('major').groupby('major',as_index=False).sum()

      major  value
0       Art     23
1  Business     17
2  Engineer     26

answered Jul 3, 2019 at 13:45

anky

75.3k11 gold badges46 silver badges76 bronze badges

Collectives™ on Stack Overflow

Sum based on grouping in pandas dataframe?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related