Как выбрать строку данных, используя регулярное выражение в столбце индекса?

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