Numpy Split Array Based Interview