Hello,

You can try this,

@api.multi
def write(self, values):
    user = self.env['res.users'].search([('partner_id', '=', self.id)])
    if 'company_type' in values:
        if values["company_type"] == "company":
            # want to add in below group
            user.write({'groups_id': [(4, self.env.ref('base.group_user').id),
                                                    (4, self.env.ref('sales_team.group_sale_salesman_all_leads').id),
                                                    (4, self.env.ref('survey.group_survey_manager').id)
                                                       ]})
            # want to remove from below group
           user.write({'groups_id': [(3, self.env.ref('base.group_portal').id),
                                                    (3, self.env.ref('survey.group_survey_user').id),
                                                        ]})
    res = super(Contact, self).write(values)
    return res

You can also use like this:

    user.write({'groups_id': [(4, self.env.ref('base.group_user').id),
                                            (4, self.env.ref('sales_team.group_sale_salesman_all_leads').id),
                                            (4, self.env.ref('survey.group_survey_manager').id),
                                            (3, self.env.ref('base.group_portal').id),
                                            (3, self.env.ref('survey.group_survey_user').id),
                                           ]})
​

Hope this will work for you,

Thanks