考虑这个小例子:
data={"X":[1, 2, 3, 4, 5], "Y":[6, 7, 8, 9, 10], "Z": [11, 12, 13, 14, 15])
frame=pd.DataFrame(data,columns=["X","Y","Z"],index=["A","A","A","B","B"])
我想分组"框架"与
grouped=frame.groupby(frame.index)
现在我想通过以下方式遍历群组:
for group in grouped:
但下一步我不管理:如何将每个循环中的组提取为pandas数据帧,这样我可以进一步处理它?</ p>
答案 0 :(得分:3)
df.groupby
返回2元组列表:索引和组。您可以像这样迭代每个组:
for _, g in frame.groupby(frame.index):
.... # do something with `g`
但是,如果要对组执行某些操作,可能有更好的方法而不是迭代。
答案 1 :(得分:2)
以下是一个例子:
groups = frame.groupby(level=0)
for n,g in groups:
print('This is group '+ str(n)+'.')
print(g)
print('\n')
输出:
This is group A.
X Y Z
A 1 6 11
A 2 7 12
A 3 8 13
This is group B.
X Y Z
B 4 9 14
B 5 10 15