Python: numpy.ndarray split