diff --git a/ProcessorComponents.py b/ProcessorComponents.py index 883cd57e4271faaaa9a6798ef82dcd6450fa78d5..138743e54fefab366b003d99e763c0a89acea2a8 100644 --- a/ProcessorComponents.py +++ b/ProcessorComponents.py @@ -1079,11 +1079,17 @@ def process_in_job_survey(jobPath,status,config,component): #TODO : for Kenya data, provide a coordinate-based filter return + def filter_by_list(df,coln,values): + # CAUTION: This requires surveyor to provide the correct list of countries + df.drop(df.loc[~df[coln].isin(values)].index,inplace=True) + return + func_types = { 'add': add_column, 'remove' : remove_column, 'replace' : replace_column, - 'filter' : filter_by_column + 'filter' : filter_by_column, + 'filter_by_list' : filter_by_list } # simple format alignment using edits on config