獲得基於在大熊貓值列表數據框的行

Question

所以，我有一個熊貓數據幀是這樣的：

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

等

，我有名字的名單，我想查詢從上面的數據框。

對於如：

names = ['foo', 'foobar'] 

我想上面的查詢數據幀，所以它返回哪些相應的行號.. 因此，例如..對應於「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')