pandas groupby取出某组数据

>>> x=pd.DataFrame({'number': [1,1,2,2],
                 'Q1': ['B0', 'B1', 'B2', 'B3'],
                 'Q2': ['C0', 'C1', 'C2', 'C3'],
                 'Q3': ['D0', 'D1', 'D2', 'D3']},)
>>> x
   number  Q1  Q2  Q3
0       1  B0  C0  D0
1       1  B1  C1  D1
2       2  B2  C2  D2
3       2  B3  C3  D3
>>> group=x.groupby('number')

有两种方法取出group中的元素
方法一:

>>> for g in group:
    	key=g[0]
    	val=g[1]
    	print('key:',key)
    	print('val:\n',val)
>>>    
key: 1
val:
    number  Q1  Q2  Q3
0       1  B0  C0  D0
1       1  B1  C1  D1
key: 2
val:
    number  Q1  Q2  Q3
2       2  B2  C2  D2
3       2  B3  C3  D3

方法二:

>>> group_1=group.get_group(1)
>>> group_1
   number  Q1  Q2  Q3
0       1  B0  C0  D0
1       1  B1  C1  D1
>>> group_2=group.get_group(2)
>>> group_2
   number  Q1  Q2  Q3
2       2  B2  C2  D2
3       2  B3  C3  D3

参考:
pandas-docs/stable/user_guide/groupby

猜你喜欢

转载自blog.csdn.net/m0_37586991/article/details/89633958
今日推荐