整理数据框,其中索引满足特定条件

时间:2016-08-19 02:08:13

标签: python pandas

我有一个这样的数据框:

          name    pe      outstanding totals     totalAssets
code                                                                      
300533   abc     30.04      2500.00   10000.00     82066.80   
300532   def     31.27      2100.00    8400.00     77945.25   
603986   NiT      23.40      2500.00   10000.00     89517.36   
600187  ITG       0.00    145562.42  145562.42    393065.88   
000652  IGE      929.15    146567.31  147557.39   2969607.50 

我想整理那些索引前3个字符的行([' 000',' 300'])

结果将是:

          name    pe      outstanding totals     totalAssets
code                                                                      
300533   abc     30.04      2500.00   10000.00     82066.80   
300532   def     31.27      2100.00    8400.00     77945.25     
000652  IGE      929.15    146567.31  147557.39   2969607.50 

感谢。

1 个答案:

答案 0 :(得分:2)

您可以使用str从索引中提取前3个字符:

df[df.index.str[:3].isin(['300', '000'])]

#       name       pe   outstanding    totals    totalAssets
#  code                 
#300533  abc    30.04       2500.00  10000.00       82066.80
#300532  def    31.27       2100.00   8400.00       77945.25
#000652  IGE   929.15     146567.31 147557.39     2969607.50