如何重複元素矩陣在MATLAB

Question

如何重複

A = [ 1 2 ; 
     3 4 ] 

反覆通過

B = [ 1 2 ; 
     2 1 ] 

因此，我希望我的回答像矩陣C：

C = [ 1 2 2; 
     3 3 4 ] 

感謝您的幫助。

Answer 1

爲了簡單起見，我假設您只會添加更多列，並且您已經檢查過每行的列數是否相同。

那麼變得重複元素和重塑的簡單組合。

編輯我修改了代碼，這樣它也可以工作，如果A和B是3D數組。

%# get the number of rows from A, transpose both 
%# A and B so that linear indexing works 
[nRowsA,~,nValsA] = size(A); 
A = permute(A,[2 1 3]); 
B = permute(B,[2 1 3]); 

%# create an index vector from B 
%# so that we know what to repeat 
nRep = sum(B(:)); 

repIdx = zeros(1,nRep); 
repIdxIdx = cumsum([1 B(1:end-1)]); 
repIdx(repIdxIdx) = 1; 
repIdx = cumsum(repIdx); 

%# assemble the array C 
C = A(repIdx); 
C = permute(reshape(C,[],nRowsA,nValsA),[2 1 3]); 

C = 
    1  2  2 
    3  3  4 

Answer 2

只是爲了好玩，另一種解決方案利用arrayfun的：

res = cell2mat(arrayfun(@(a,b) ones(b,1).*a, A', B', 'uniformoutput', false))' 

這導致：

res = 

    1  2  2 
    3  3  4