matlab - 在 MATLAB 中置换矩阵的列

Question

假设我有一个 n by d 矩阵 A 并且我想置换某些列的条目。为此，我将 1 ... n 的排列计算为

idx1 = randperm(n)'
idx2 = randperm(n)'

然后我可以这样做:

A(:,1) = A(idx1,1)
A(:,2) = A(idx2,2)

但是，我不想使用 for 循环来执行此操作，因为它会很慢。假设我有一个 n by d 矩阵 A 和一个 n by d 索引矩阵 IDX 指定排列，是否有以下 for 循环的更快等价物:

for i = 1:d
    A(:,i) = A(IDX(:,i),i);
end

Answer 1

使用 linear-indexing在bsxfun的帮助下-

[n,m] = size(A);
newA = A(bsxfun(@plus,IDX,[0:m-1]*n))