Advertisement
Dodma

Untitled

Jul 10th, 2020
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.68 KB | None | 0 0
  1. from pymystem3 import Mystem
  2. m = Mystem()
  3. rests['object_name_lemma'] = rests['object_name'].apply(m.lemmatize)
  4. def categorizer(row):
  5. if 'школа' in row:
  6. return 'школьная столовая'
  7. if 'гоу' in row:
  8. return 'школьная столовая'
  9. if 'при' in row:
  10. return 'школьная столовая'
  11. if 'столовая для персонала' in row:
  12. return 'школьная столовая'
  13. if 'гбоу' in row:
  14. return 'школьная столовая'
  15. if 'шк' in row:
  16. return 'школьная столовая'
  17. if 'столовая при колледже' in row:
  18. return 'школьная столовая'
  19. if 'комбинат' in row:
  20. return 'школьная столовая'
  21. if 'школьно' in row:
  22. return 'школьная столовая'
  23. if 'закрытое' in row:
  24. return 'школьная столовая'
  25. if 'теремок' in row:
  26. return 'теремок'
  27. if 'kfc' in row:
  28. return 'kfc'
  29. if 'шаурма' in row:
  30. return 'шаурма'
  31. if 'братья караваевы' in row:
  32. return 'братья караваевы'
  33. if 'шоколадница' in row:
  34. return 'шоколадница'
  35. if 'домино' in row:
  36. return 'домино пицца'
  37. if 'мята' in row:
  38. return 'мята'
  39. if 'макдоналдс' in row:
  40. return 'макдоналдс'
  41. if 'бургер кинг' in row:
  42. return 'бургер кинг'
  43. rests['object_name'] = rests['object_name_lemma'].apply(categorizer)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement