Получение неупорядоченного индекса с учетом соответствующих критериев

Question

У меня есть кадр данных pandas, который уже отфильтрован, поэтому индексы расположены не по порядку (т. Е. Не от 0 до конца моего кадра данных).У меня есть колонка, которая является «мешком слов».Это просто список слов.Я знаю слово, которое я ищу.Как я могу найти индекс / индексы, которые содержат это слово?
Я пытался использовать функцию '.index', но я не могу использовать его правильно.

Пример DataFrame

Желаемым выводом будет просто индекс / индексы записи / записей, которые содержат слово, которое я ищу.

Answer 1

Простой хак для вашего случая, рассматривая столбец сумки как строку, а не список, если вам нужно проверить, содержит только

df['word_flag'] = df['bag'].astype(str).str.contains('your-word-here')

Если вам конкретно нужны индексы

df[df['word_flag'] == 1].index