def categorize_purpose(row): if 'автом' in row['purpose']: return 'операции с автомобилем' if 'недвиж' in row['purpose']: return 'операции с недвижимостью' if 'свадьб' in row['purpose']: return 'проведение свадьбы' if 'образов' in row['purpose']: return 'получение образования' data['purpose_category'] = data.apply(categorize_purpose, axis=1)