Create function for filtering in ntb.

Author: MarenBermudezBoeckle

src/project_mbb/config.py

@@ -16,7 +16,7 @@
 DOCUMENTS = ROOT.joinpath("documents").resolve()
 
 
-VARIABLES = [
+REL_VAR = [
     "enc_rpc",
     "rph_ID",
     "idhogar",
@@ -25,4 +25,88 @@
     "rph_nivel",
     "rph_ciuo",
     "rph_nse",
+    "rph_parentesco",
+    "rph_edad",
+    "rph_sexo",
+    "rph_idgen",
+    "Kish",
+    "Hogar_Kish",
+    "ANTIG_SECTOR",
+    "Fact_Pers_Com",
+    "Fact_Pers_Reg",
+    "Fact_Pers_Regional_102",
+    "Fact_Hog_Com",
+    "Fact_Hog_Reg",
+    "Fact_Hog_Regional_102",
+    "VarStrat",
+    "Conglomerado",
+    "P_AUMENTO_PAIS",
+    "P_AUMENTO_COM",
+    "P_AUMENTO_BARRIO",
+    "P_FUENTE_INFO_PAIS_1",
+    "P_FUENTE_INFO_COM_1",
+    "P_INSEG_LUGARES_1",
+    "P_INSEG_LUGARES_2",
+    "P_INSEG_OSCURO_1",
+    "P_INSEG_DIA_1",
+    "P_INSEG_OSCURO_2",
+    "P_INSEG_DIA_2",
+    "P_DESORDENES_3",
+    "P_INCIVILIDADES_6",
+    "P_EXPOS_DELITO",
+    "P_DELITO_PRONOSTICO__1",
+    "P_DELITO_PRONOSTICO__2",
+    "P_DELITO_PRONOSTICO__3",
+    "P_DELITO_PRONOSTICO__5",
+    "P_DELITO_PRONOSTICO__6",
+    "P_DELITO_PRONOSTICO__77",
+    "P_DELITO_PRONOSTICO__88",
+    "P_DELITO_PRONOSTICO__99",
+    "P_MOD_ACTIVIDADES_1",
+    "P_MOD_ACTIVIDADES_3",
+    "P_MOD_ACTIVIDADES_8",
+    "P_MOD_ACTIVIDADES_7",
+    "P_MOD_ACTIVIDADES_11",
+    "EV_CONOCE_CCH",
+    "EV_CONFIA_CCH",
+    "EV_CONOCE_PDI",
+    "EV_CONFIA_PDI",
+    "EV_CONOCE_FMP",
+    "EV_CONFIA_FMP",
+    "HUR_QUIEN__0",
+    "HUR_QUIEN__1",
+    "HUR_QUIEN__2",
+    "HUR_QUIEN__3",
+    "HUR_QUIEN__4",
+    "HUR_QUIEN__5",
+    "HUR_QUIEN__6",
+    "HUR_QUIEN__7",
+    "HUR_MES",
+    "HUR_DONDE_ESP",
+    "HUR_DONDE_COMUNA",
+    "HUR_DONDE_REGION",
+    "HUR_DENUNCIA",
+    "HUR_DENUNCIA_COMO",
+    "HUR_SATISF_DEN",
+    "HUR_ACCION_POLICIAL",
+    "HUR_CONTACTO_MP",
+    "HUR_MOTIV_NO_DEN",
+    "SCREEN_ROB_HUR",
+    "HUR_DENUNCIAS",
+]
+
+AUX_VAR = [
+    "PAD",
+    "PADC",
+    "PADB",
+    "PCOS",
+    "VH_DV",
+    "VP_DV",
+    "DEN_VHDV",
+    "COSC_DV",
+    "DEN_ROBOS",
+    "VH_ROBOS",
+    "RDV",
+    "RDV_INTENTO",
+    "HUR",
 ]

src/project_mbb/data_management/task_data_management.py

@@ -1,8 +1,8 @@
 import pickle
+from pathlib import Path
 
 import pandas as pd
 import pyreadstat
-from Pathlib import Path
 
 from project_mbb.config import BLD, DATA