Python Numpy Split Array Into Two