Numpy Split Array Based Stack Codechef Problems