The question is how to implement a conditional in numpy?