Reordering matrix with a permutation vector but keeping the original size of matrix Reordering matrix with a permutation vector but keeping the original size of matrix numpy numpy