Sorting Numpy Array Interview