Использование lambda условных и pandas str.contains для комков строк

Question

Попытка узнать кое-что, я возился с глобальной базой атак акул на Kaggle, и я пытаюсь найти лучший способ разбить строки, используя функцию lambda и str.contains.

В любом случае строка содержит фразу с skin diving, например. 'skin diving for abalone', в столбце data['Activity'] Я хочу заменить действие skin diving. (Есть 92 вариантов для подводной охоты, следовательно, пытаются использовать функцию лямбды)

я могу вернуть булеву серию, используя

data['Activity].str.contains('skin diving') 

Но я не уверен, как изменить значение, если это условие истинно

Моя функция лямбда = data.apply(lambda x: 'free diving' if x.str.contains('free diving)), но я получаю синтаксическую ошибку, и я недостаточно хорошо знаком с функциями лямбда и пандами, чтобы получить ее правильно, любая помощь будет оценена.

Answer 1

Вместо того, чтобы использовать метод Series.str, вы можете использовать в оператора в вашем лямбда для проверки подстроки

data['activity'] = data['activity'].apply(lambda x: 'skin diving' if 'skin diving' in x else x) 

Answer 2

Вы можете использовать метод str.containsnp.where с

In [141]: df 
Out[141]: 
     activity 
0 free diving ok 
1    ok 

In [142]: df.activity = np.where(df.activity.str.contains('free diving'), 
           'free diving', df.activity) 

In [143]: df 
Out[143]: 
     activity 
0 free diving 
1   ok