如何在matlab中重复元素矩阵

Question

如何重复

A = [ 1 2 ; 
      3 4 ]

重复

B = [ 1 2 ; 
      2 1 ]

所以我想要像矩阵 C 一样的答案：

C = [ 1 2 2; 
      3 3 4 ]

感谢您的帮助。

Answer 1

只是为了好玩，另一个使用arrayfun的解决方案：

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

这会导致：

res =

     1     2     2
     3     3     4

Answer 2

为了简单起见，我假设您只是要添加更多列，并且您已经检查了每行的列数是否相同。

然后它变成了重复元素和重塑的简单组合。

编辑我已经修改了代码，因此如果 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