Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def sorting_lemmas(lemmas_type):
- """
- проверяет столбец 'lemmas_type' на наличие заданных слов, при выполнении условия
- возвращает заданные значения
- """
- if ('жилье' in lemmas_type) or ('недвижимость' in lemmas_type):
- return 'недвижимость'
- if 'автомобиль' in lemmas_type:
- return 'автомобиль'
- if 'свадьба' in lemmas_type:
- return 'свадьба'
- else:
- return 'образование'
- data['target_credit'] = data['lemmas_type'].apply(sorting_lemmas)
- print(data.head(5))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement