Понимание списка, если ключевое слово в строке ⇐ Python

[Anonymous]

Я очищаю столбцы в кадре данных и хотел бы разделить столбец «описание» на основе определенных ключевых слов. Я написал одну строку, но по какой-то причине все равно возвращает np.nan, даже если условие выполнено.

Код: Выделить всё

def find_and_suck_hashtag_vacuum(string, keyword, output):
return [output if keyword in string else np.nan]

с переменными, установленными как

Код: Выделить всё

string_samp = ['Air conditioning,Car park,Interesting wine list,Restaurant offering vegetarian menus,Wheelchair access']

output = find_and_suck_hashtag_vacuum(string_samp, "Air", "AC")

Но в приведенном выше случае функция все равно возвращает np.nan. Что мне следует изменить?

Подробнее здесь: https://stackoverflow.com/questions/790 ... -in-string