Matlab中的无环子矩阵赋值

Question

我有一个大小为D-by-N 的矩阵F 和一个长度为N 的向量N，由[1,a] 范围内的随机整数组成。我想创建一个大小为D * a 的矩阵M，使得每个列M(:,i) 具有从索引(A(i)-1)*D+1 到(A(i)-1)*D+D 的向量F(:,i)。

例子：

F = [1 2 3 10
     4 5 6 22]

A = [3 2 1 2]

a = 4

M = [0 0 3 0
     0 0 6 0
     0 2 0 10
     0 5 0 22
     1 0 0 0
     4 0 0 0
     0 0 0 0
     0 0 0 0]

我可以用一个简单的循环来做到这一点

        for i = 1 : N
            M((A(i)-1)*D+1:(A(i)-1)*D+D,i) = F(:,i);
        end

但对于大型N，这可能需要一段时间。我正在寻找一种没有循环的方法。

Answer 1

您可以将bsxfun 用于基于linear-indexing 的方法-

[D,N] = size(F);    %// Get size of F

start_idx = (A-1)*D+1 + [0:N-1]*D*a;          %// column start linear indices
all_idx = bsxfun(@plus,start_idx,[0:D-1]');   %//'# all linear indices

out = zeros(D*a,N); %// Initialize output array with zeros
out(all_idx) = F;   %// Insert values from F into output array

示例运行 -

F =
     1     2     3    10
     4     5     6    22
A =
     3     2     1     2
a =
     4
out =
     0     0     3     0
     0     0     6     0
     0     2     0    10
     0     5     0    22
     1     0     0     0
     4     0     0     0
     0     0     0     0
     0     0     0     0