这类似于MultiIndex-based indexing in pandas。
有没有更好的方法来迭代子系列?
df = pd.DataFrame([[1,1,1], [1,2,1], [1,2,2],
[2,1,1], [2,2,1], [2,3,1], [2,3,2], [2,3,3]],
columns=['a', 'b', 'c'])
g = df.groupby(['a', 'b']).size()
for label in g.index.levels[0]:
print(label)
print(g[label])
这将给出:
1
b
1 1
2 2
dtype: int64
2
b
1 1
2 1
3 3
dtype: int64
像这样的伪代码:
for label, series in g.get_sub_series(level = 0):
print(label)
print(series)