Получить индекс значений, соответствующих условию, используя панд

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 (правильный ли это аргумент для щенка?), А также эти вопросы SO:

Поиск местоположения метки в индексе DataFrame

Поиск индекса элемента в pandas Series

Код работает нормально, когда все, что я прошу сделать, это добавить 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()