python - 按级别从多索引分组中获取第一个索引

Question

我有一个多索引 Pandas 数据框:

                              SHOPPING_COUNT  
CLIENT          YEAR MONTH               
1000063         2013 12             9  
                2014 1              9  
                     2              7  
                     3              9  
                2015 4              6  
                     5              5  
                     6              9  
1001327         2014 5              1  
                     6              1  
                2015 2              7  
                     3              1  
                     4              3 
1001399         2013 8              1 

我想知道每个客户端的第一个索引，按级别 0 排序。我的意思是，我想得到:

1000063         2013 12
1001327         2014 5
1001399         2013 8              

Answer 1

让 df 成为你的数据框，你可以这样做:

df = df.groupby(level=0).apply(lambda x: x.iloc[0:1])
df.index = df.index.droplevel(0)

实际上这应该更容易做到，但我认为这种方法有效。