我有一个数据框,如下所示。我想根据列 ID 从该数据帧创建多个数据帧。
df = pd.DataFrame(results)
print(df)
结果是:
ID NAME COLOR
0 01 ABC RED
1 01 ABC ORANGE
2 01 ABC WHITE
3 02 DEF RED
4 02 DEF PURPLE
5 02 DEF GREEN
6 02 DEF ORANGE
7 02 DEF BLACK
8 03 GHI RED
9 03 GHI BLACK
10 03 GHI GREEN
11 03 GHI ORANGE
12 04 JKL RED
多个数据框应如下所示:我无法将其放入 python 代码中,请帮助。
ID NAME COLOR
0 01 ABC RED
1 01 ABC ORANGE
2 01 ABC WHITE
ID NAME COLOR
0 02 DEF RED
1 02 DEF PURPLE
2 02 DEF GREEN
3 02 DEF ORANGE
4 02 DEF BLACK
ID NAME COLOR
0 03 GHI RED
1 03 GHI BLACK
2 03 GHI GREEN
3 03 GHI ORANGE
ID NAME COLOR
0 04 JKL RED
你可以做:
data_dict={'df'+str(i): grp for i , grp in df.groupby('ID')}
其中给出了一本字典:
{'df1': ID NAME COLOR
0 1 ABC RED
1 1 ABC ORANGE
2 1 ABC WHITE, 'df2': ID NAME COLOR
3 2 DEF RED
4 2 DEF PURPLE
5 2 DEF GREEN
6 2 DEF ORANGE
7 2 DEF BLACK, 'df3': ID NAME COLOR
8 3 GHI RED
9 3 GHI BLACK
10 3 GHI GREEN
11 3 GHI ORANGE, 'df4': ID NAME COLOR
12 4 JKL RED}
现在只需调用每个键即可访问每组ID,
print(data_dict['df2'])
ID NAME COLOR
3 2 DEF RED
4 2 DEF PURPLE
5 2 DEF GREEN
6 2 DEF ORANGE
7 2 DEF BLACK
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)