Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- df = pd.DataFrame({
- 'Part': [1,1,1,3,3,2,2,2],
- 'other': ['a','b','c','d','e','f','g','h']
- })
- d = df.groupby('Part').apply(lambda d: d.to_dict('records')).to_dict()
- print d
- {1: [{'Part': 1, 'other': 'a'},
- {'Part': 1, 'other': 'b'},
- {'Part': 1, 'other': 'c'}],
- 2: [{'Part': 2, 'other': 'f'},
- {'Part': 2, 'other': 'g'},
- {'Part': 2, 'other': 'h'}],
- 3: [{'Part': 3, 'other': 'd'}, {'Part': 3, 'other': 'e'}]}
- df = pd.DataFrame({"Part": [1, 1, 2, 2],
- "Var1": [10, 11, 12, 13],
- "Var2": [20, 21, 22, 23]})
- dfg = df.groupby("Part")
- df1 = dfg.get_group(1)
- df2 = dfg.get_group(2)
- for grp in dfg.groups:
- print(dfg.get_group(grp))
- print()
- Part Var1 Var2
- 0 1 10 20
- 1 1 11 21
- Part Var1 Var2
- 2 2 12 22
- 3 2 13 23
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement