def get_region_columns()

in dataset_prep.py [0:0]

• 2 lines of code
• 3 McCabe index (conditional complexity)

def get_region_columns(df, region):
    return [col for col in df.columns if region in col]