Numpy array is 2d with indices splitting the data indices