how do i split an ndarray based on array of indexes?,"print(np.split(a, b, axis=0))"