使用 pandas 获取与条件匹配的值的索引

Question

我有一本字典wrong_names、一个熊猫系列df['name']和一个空列表i_list。

我想遍历df['name']，如果名称在 中wrong_names.keys()，我想将该元素的索引附加df到i_list. 这是我到目前为止所拥有的：

i_list = []
for pup in df['name']:
    if pup in wrong_names.keys():
    i_list.append(df.index.get_loc(pup))
i_list

我已经看过 .get_loc 文档（pup 是正确的论点吗？），以及这些 SO 问题：

在 DataFrame 索引中查找标签位置

在熊猫系列中查找元素的索引

当我要求它做的只是附加pup到 时i_list，代码工作正常，所有迹象表明循环正确识别何时pup也是字典中的键，

i_list = []
for pup in df['name']:
    if pup in wrong_names.keys():
    i_list.append(pup)
i_list

但我无法获得索引列表，这是我下一步所需要的。提前致谢。

Answer 1

• 不需要迭代。
• 使用Series.isin对系列成员检查
• 使用结果屏蔽，以索引df.index和结果转换到一个列表。

i_list = df.index[df['name'].isin(wrong_names)].tolist()