Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from pymystem3 import Mystem
- m = Mystem()
- rests['object_name_lemma'] = rests['object_name'].apply(m.lemmatize)
- def categorizer(row):
- if 'школа' in row:
- return 'школьная столовая'
- if 'гоу' in row:
- return 'школьная столовая'
- if 'при' in row:
- return 'школьная столовая'
- if 'столовая для персонала' in row:
- return 'школьная столовая'
- if 'гбоу' in row:
- return 'школьная столовая'
- if 'шк' in row:
- return 'школьная столовая'
- if 'столовая при колледже' in row:
- return 'школьная столовая'
- if 'комбинат' in row:
- return 'школьная столовая'
- if 'школьно' in row:
- return 'школьная столовая'
- if 'закрытое' in row:
- return 'школьная столовая'
- if 'теремок' in row:
- return 'теремок'
- if 'kfc' in row:
- return 'kfc'
- if 'шаурма' in row:
- return 'шаурма'
- if 'братья караваевы' in row:
- return 'братья караваевы'
- if 'шоколадница' in row:
- return 'шоколадница'
- if 'домино' in row:
- return 'домино пицца'
- if 'мята' in row:
- return 'мята'
- if 'макдоналдс' in row:
- return 'макдоналдс'
- if 'бургер кинг' in row:
- return 'бургер кинг'
- rests['object_name'] = rests['object_name_lemma'].apply(categorizer)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement