Найдите строку в одном столбце и извлеките весь столбец для соответствующей строки Python

Question

Я пытаюсь найти строку в одном столбце и хочу извлечь весь фрейм данных, когда строка соответствует столбцу

Мои данные выглядят следующим образом

**string**  **Number**
hello there     11
Hello hi        22
How are you     33

Мой код следующим образом

string_hello = []
for i in df['string']:
    if re.search("Hello",i, flags=re.I): 
        string_hello.append(i)

Ожидаемый результат:

**string**  **Number**
hello there     11
Hello hi        22

Но мой код дает только строковый столбец, как я могу извлечь числовой столбец тоже?

Answer 1

Вы можете использовать str.contains и выполнить логическое индексирование на фрейме данных с результатом:

df[df.string.str.lower().str.contains(r'hello')]

      string    Number
0  hello there      11
1     Hello hi      22