Numpy Array Logical Interview