Slice Numpy Array Based Interview Question