python - MultiIndexed groupby 的访问键

Question

我有一个熊猫 groupby 对象，来自两个键。

gb = df.groupby(['A','B'])

我如何访问特定的密钥说(2,4)，我该怎么做？如果只有一个键，
该方法效果很好。group_by()

有任何想法吗？

Answer 1

我想你正在寻找get_group：

In [1]: df = pd.DataFrame([[2, 4, 1], [2, 4, 2], [3, 4, 1]], columns=['A', 'B', 'C'])

In [2]: df
Out[2]:
   A  B  C
0  2  4  1
1  2  4  2
2  3  4  1

In [3]: g = df.groupby(['A', 'B'])

In [4]: g.get_group((2,4))
Out[4]:
   A  B  C
0  2  4  1
1  2  4  2

Answer 2

使用元组get_group

In [49]: df = DataFrame(np.random.randint(10,size=15).reshape(5,3),columns=list('ABC'))

In [50]: df
Out[50]: 
   A  B  C
0  8  9  2
1  7  5  3
2  3  1  2
3  2  4  0
4  6  9  4

In [51]: df.groupby(['A','B']).sum()
Out[51]: 
     C
A B   
2 4  0
3 1  2
6 9  4
7 5  3
8 9  2

In [52]: df.groupby(['A','B']).get_group((6,9))
Out[52]: 
   A  B  C
4  6  9  4