df1 = pd.DataFrame({'KEY':['C','B','C','A','B','B','A','C','A'],
'DATA':[2,4,6,8,10,1,14,16,18]})
df1
KEY DATA
0 C 2
1 B 4
2 C 6
3 A 8
4 B 10
5 B 1
6 A 14
7 C 16
8 A 18
gp = df1.groupby(by='KEY') #通过KEY进行分组
<pandas.core.groupby.generic.DataFrameGroupBy object at 0x00000296D65FEB08>
for i in gp: # 通过groupby分组后的数据是DataFrameGroupBy类型,需要遍历
print(i)
('A', KEY DATA
3 A 8
6 A 14
8 A 18)
('B', KEY DATA
1 B 4
4 B 10
5 B 1)
('C', KEY DATA
0 C 2
2 C 6
7 C 16)