Numpy Array Get Index Of Conditional Statement In Python