Question

我有一个pandas 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

访问MultiIndex组的密钥

2 个答案: