получение строк dataframe на основе списка значений в панд

Question

Итак, у меня есть панд dataframe вроде этого:

name, val1, val2 
foo, 1.2, 1.7 
bar, 2.1, 21 
baz, 9, 21 
foobar, 14, 29 

и так далее

и у меня есть список имен, которые я хочу запросить из вышеуказанной рамки данных.

Для например:

names = ['foo', 'foobar'] 

Я хочу запросить выше dataframe, так что она возвращает то, что есть соответствующие номера строк .. Так, например .. индексы, соответствующие «Foo» и «Foobar» 0 и 3?

Как, я могу достичь этого?

Answer 1

вариант 1
использование query

df.query('name in @names').index 

Int64Index([0, 3], dtype='int64') 

вариант 2
использование isin

df.index[df.name.isin(names)] 

Int64Index([0, 3], dtype='int64')