根据熊猫中的值列表获取数据框的行

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