Numpy Split Array Based Stack Implementation In Python