def formula(individu, period, parameters): age = individu('age', period) apprentissage_contrat_debut = individu('apprentissage_contrat_debut', period) smic = parameters(period).marche_travail.salaire_minimum.smic.smic_b_horaire * 52 * 35 / 12 anciennete_contrat = ( datetime64(period.start) + timedelta64(1, 'D') - apprentissage_contrat_debut ).astype('timedelta64[Y]') apprenti = individu('apprenti', period) salaire_en_smic = [ # TODO: move to parameters dict( part_de_smic_by_anciennete = { 1: .25, 2: .41, 3: .53, }, age_min = 15, age_max = 18, ), dict( part_de_smic_by_anciennete = { 1: .37, 2: .49, 3: .61, }, age_min = 18, age_max = 21, ), dict( part_de_smic_by_anciennete = { 1: .53, 2: .65, 3: .78, }, age_min = 21, age_max = 99 ) ] output = age * 0.0 for age_interval in salaire_en_smic: age_condition = (age_interval['age_min'] <= age) * (age < age_interval['age_max']) output[age_condition] = sum([ (anciennete_contrat[age_condition] == timedelta64(anciennete, 'Y')) * part_de_smic for anciennete, part_de_smic in age_interval['part_de_smic_by_anciennete'].items() ]) return output * smic * apprenti