Split Numpy Array Based Interview