使用熊猫仅获取n个索引的值数量

Question

我的系列赛很流畅

               Data
Level1 Level2
A        A1      10
         A2      11
         A3      12
         A4      13
B        B1      14
         B2      15
         B3      16
         B4      17
         B5      18

我希望在保持级别1不变的情况下，仅获得级别2的前三个条目。

             Data
Level1 Level2
A        A1      10
         A2      11
         A3      12
B        B1      14
         B2      15
         B3      16

我尝试使用iloc或loc或indexslice，但没有使用正确的语法。我是熊猫新手。这是一门课程的一项作业，我花了更多时间进行尝试，但无法弄清楚。

Answer 1

在第一级使用GroupBy.head：

df1 = df.groupby(level=0).head(3)

如果使用pandas 0.24+：

df1 = df.groupby('Level1').head(3)

Answer 2

df.groupby('Level1').head(3).reset_index(drop=True)