Python: Elegant and efficient ways to mask a list

Question

Example:

from __future__ import division
import numpy as np

n = 8
\"\"\"masking lists\"\"\"
lst = range(n)
print lst

# the mask (filter)         


        

Answer 1

You could also just use list and zip

1. define a funcion

def masklist(mylist,mymask):
    return [a for a,b in zip(mylist,mymask) if b]

2. use it!

n = 8
lst = range(n)
msk = [(el>3) and (el<=6) for el in lst]
lst_msk = masklist(lst,msk)
print(lst_msk)