获得基于在大熊猫值列表数据框的行

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')