获取子数据帧的最简单方法是什么？

Question

x是一个数据框，其索引是代码，包含一列pe。

>>>x
     pe  
code                                                                     
01   15
02   30
03   70
04   6
05   40
06   34
07   25
08   65
10   45
12   55
13   32

获取x的索引。

x.index
Index(['01', '02', '03', '04', '05', '06', '07',
       '08', '10', '12','13'],
      dtype='object', name='code', length=11)

我希望获得一个索引为['01'，'04'，'08'，'10'，'12']的子数据帧。

x_sub
        pe  
code                                                                     
01   15
04   6
08   65
10   45
12   55

获取子数据帧的最简单方法是什么？

Answer 1

使用loc

x_sub = x.loc[['01','04','08','10','12']]

或者如果索引是整数：

x_sub = x.loc[[01,04,08,10,12]]