The original code already look crap due to making a new list containing object instead of a mask or something else. Also that can be done using list comprehensi

The original code already look crap due to making a new list containing object instead of a mask or something else. Also that can be done using list comprehension. Also it totally can be vectorize or parallelize. You need more experienced Python engineer.