from pymystem3 import Mystem
m = Mystem()
def category(data):
    for row in data:
        row = str(row)
        lemma = m.lemmatize(row)
        if 'жиль' in lemma:
            return 'жилье'
        if 'недвиж' in lemma:
            return 'недвижимость'
        if 'свадь' in lemma:
            return 'свадьба'
        if 'образован' in lemma:
            return 'образование'
        return 'авто'
df['category'] = df['purpose'].apply(category)