from pymystem3 import Mystem
m = Mystem()

def purpose_group(row):
    row_str = str(row)
    lemmas = m.lemmatize(row_str)
    if  'недвижимость'in lemmas or 'жилье' in lemmas:
        return 'недвижимость'
    if 'автомобиль'in lemmas:
        return 'автомобиль'
    if 'свадьба'in lemmas:
        return 'свадьба'
    return 'образование'
solvency['purpose_group'] = solvency['purpose'].apply(purpose_group)
solvency.head(20)