2015-09-25 53 views
1

如何获得分组的项目,我有:使用GROUPBY方法

df = pd.DataFrame({'col1':['A','A','A','B','B','B'], 'col2':['C','D','D','D','C','C'], 'col3':[.1,.2,.4,.6,.8,1]}) 

然后我组两列如下:

g = df.groupby(['col2','col1']) 

然后,我想获得所有组的名字的列表(例如像g.groupnames())和:

g.get_group('D','A') 

回答

2
g.groups.keys() 

[( 'C', 'A'),( 'C', 'B'),( 'd', 'A'),( 'd', 'B')]

g.groups[('D','A')] 

[ 1,2]

如果你想使用g.get_group(),你需要传递一个元组,而不是两个字符串:

g.get_group(('D','A'))