Numpy Index 2d Array With 1d Array