Advertisement
Guest User

Untitled

a guest
Mar 18th, 2019
57
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.39 KB | None | 0 0
  1. #INPUT
  2. # 1-A 1-B 2-A 5-B
  3. #0 1 3 3 2
  4. #1 2 4 4 7
  5.  
  6. #1-A: person id is 1, expense category is A
  7. #rows are months
  8. #calculate total expense per person per month
  9.  
  10. #OUTPUT
  11. # 1 2 5
  12. #0 4 3 2
  13. #1 6 4 7
  14.  
  15.  
  16. import pandas as pd
  17.  
  18. d = {'1-A': [1, 2], '1-B': [3, 4], '2-A': [3, 4], '5-B': [2, 7]}
  19. df = pd.DataFrame(data=d)
  20.  
  21. df1 = df.groupby(lambda x: x.split('-')[0], axis=1).sum()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement