Loopless Назначение подматрица в Matlab

Question

У меня есть матрица F размера D-by-N и вектор A длины N случайных чисел в диапазоне [1,a]. Я хочу создать матрицу M размера D * a так, чтобы каждый колокол M(:,i) имел вектор F(:,i), начиная с индекса (A(i)-1)*D+1 до (A(i)-1)*D+D.

Пример:

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