def formula(menage, period): ''' Retrouve la zone APL (aide personnalisée au logement) de la commune en fonction du depcom (code INSEE) ''' depcom = menage('depcom', period) preload_zone_apl() default_value = 2 zone = fromiter( ( zone_apl_by_depcom.get(depcom_cell if isinstance(depcom_cell, str) else depcom_cell.decode('utf-8'), default_value) for depcom_cell in depcom ), dtype = int16, ) return select( (zone == 1, zone == 2, zone == 3), # The .index is not striclty necessary, but it improves perfomances by avoiding a later encoding (TypesZoneApl.zone_1.index, TypesZoneApl.zone_2.index, TypesZoneApl.zone_3.index) )