如何使用groupby方法获取分组项目

Question

我有：

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')

Answer 1

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'))