python - Pandas 仅使用该 key 的一部分访问组元组 key 的长度

Question

我有一个 DataFrameGroupBy，其键的结构为 Hour, ID。

我正在尝试获取每小时每个键的每个组的大小。

运行 mygroup.size() 给出的输出如下:

   ID
0  41           3
   55          10
   56           1
   60           7
   65           1
...
23  2218         5
    2222         9
    2223         5
    2225         2

我想要做的是过滤此列表，以便我可以根据键的 Hour 部分 (0-23) 获取每个组中的总数

Answer 1

调用count并传递 level=0，例如:

In [21]:
df = pd.DataFrame({'a':[0,0,1,1,1,1],'b':[1,2,3,12,3,4],'c':np.arange(6)})
df

Out[21]:
   a   b  c
0  0   1  0
1  0   2  1
2  1   3  2
3  1  12  3
4  1   3  4
5  1   4  5

In [22]:    
gp = df.groupby(['a','b'])
gp.size()

Out[22]:
a  b 
0  1     1
   2     1
1  3     2
   4     1
   12    1
dtype: int64

In [23]:
gp.size().count(level=0)

Out[23]:
a
0    2
1    3
dtype: int64