使用 MultiIndex 迭代 DataFrame

給定以下 DataFrame:

In [11]: df = pd.DataFrame({'a':[1,1,1,2,2,3],'b':[4,4,5,5,6,7,],'c':[10,11,12,13,14,15]})

In [12]: df.set_index(['a','b'], inplace=True)

In [13]: df
Out[13]: 
      c
a b    
1 4  10
  4  11
  5  12
2 5  13
  6  14
3 7  15

你可以按 MultiIndex 的任何級別進行迭代。例如,level=0(你也可以按名稱選擇級別,例如 level='a'):

In[21]: for idx, data in df.groupby(level=0):
            print('---')
            print(data)      c
a b    
1 4  10
  4  11
  5  12      c
a b    
2 5  13
  6  14      c
a b    
3 7  15

你還可以按名稱選擇級別,例如`level =‘b’:

In[22]: for idx, data in df.groupby(level='b'):
            print('---')
            print(data)
---     
      c
a b    
1 4  10
  4  11      c
a b    
1 5  12
2 5  13      c
a b    
2 6  14      c
a b    
3 7  15