Indexing 3d numpy array with 2d array Indexing 3d numpy array with 2d array numpy numpy