Advertisement
Guest User

Untitled

a guest
Jun 26th, 2019
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.83 KB | None | 0 0
  1. df = pd.DataFrame({
  2. 'Part': [1,1,1,3,3,2,2,2],
  3. 'other': ['a','b','c','d','e','f','g','h']
  4. })
  5.  
  6. d = df.groupby('Part').apply(lambda d: d.to_dict('records')).to_dict()
  7. print d
  8.  
  9. {1: [{'Part': 1, 'other': 'a'},
  10. {'Part': 1, 'other': 'b'},
  11. {'Part': 1, 'other': 'c'}],
  12. 2: [{'Part': 2, 'other': 'f'},
  13. {'Part': 2, 'other': 'g'},
  14. {'Part': 2, 'other': 'h'}],
  15. 3: [{'Part': 3, 'other': 'd'}, {'Part': 3, 'other': 'e'}]}
  16.  
  17. df = pd.DataFrame({"Part": [1, 1, 2, 2],
  18. "Var1": [10, 11, 12, 13],
  19. "Var2": [20, 21, 22, 23]})
  20. dfg = df.groupby("Part")
  21.  
  22. df1 = dfg.get_group(1)
  23. df2 = dfg.get_group(2)
  24.  
  25. for grp in dfg.groups:
  26. print(dfg.get_group(grp))
  27. print()
  28.  
  29. Part Var1 Var2
  30. 0 1 10 20
  31. 1 1 11 21
  32.  
  33. Part Var1 Var2
  34. 2 2 12 22
  35. 3 2 13 23
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement