Pandas Dataframe, aggregate and put the data in the next column

Question

I have a dataset like this:

Country   Name      Match   Result
US        Martin    Win     3
US        Martin    Lose    1
US        Martin    Draw    5
UK        Luther    Win     5
UK        Luther    Draw    3

I'd like to add two more columns with sum result from Win, Lose and Draw, and percentage of each match like this:

Country   Name      Match   Result  All Percentage
US        Martin    Win     3       8   0.375
US        Martin    Lose    1       8   0.125
US        Martin    Draw    5       8   0.625
UK        Luther    Win     6       10  0.6
UK        Luther    Draw    4       10  0.4

I've already tried using groupby and got result for size total match. However I don't know how to put it in the next column.

Thank you

Shouldn't the All column for Martin be 9, not 8 (5 + 3 + 1)? — boot-scootin
– boot-scootin, Commented Jan 4, 2017 at 14:51

jezrael · Accepted Answer · 2017-01-04 14:54:36Z

1

IIUC you need GroupBy.transform, sample DataFrame was changed:

df['All'] = df.groupby(['Country','Name'])['Result'].transform('sum')
df['Percentage'] = df.Result.div(df.All)
print (df)
  Country    Name Match  Result  All  Percentage
0      US  Martin   Win       2    8       0.250
1      US  Martin  Lose       1    8       0.125
2      US  Martin  Draw       5    8       0.625
3      UK  Luther   Win       6   10       0.600
4      UK  Luther  Draw       4   10       0.400

answered Jan 4, 2017 at 14:54

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

Pandas Dataframe, aggregate and put the data in the next column

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