如何在索引列上使用正则表达式选择数据帧行？

Question

我是熊猫新手。 下面是一个例子的问题

df = pd.DataFrame(data={'id':['john','joe','zack']})

我知道我可以像这样选择“id”列包含“jo”的行

mask = df['id'].str.contains('jo')
df[mask]

但是假设id列被索引了

df = df.set_index('id')

现在如何选择索引列包含“jo”的行？你知道吗

Answer 1

您需要将id更改为index：

df = pd.DataFrame(data={'id':['john','joe','zack'],
                        'col':[1,2,3]})

df = df.set_index('id')
df1 = df[df.index.str.contains('jo')]
print (df1)
      col
id       
john    1
joe     2