对于R2015a和更新……
根据repelem的文档(在R2015a版本中首次引入),它也可以在矩阵上运行.我相信以下代码应该完成你想要的(我无法测试它,因为我有一个旧版本):
newMat = repelem(mat, 1, 1, [3 2 1 1 5 4 2 2 1]);
对于R2015a之前的版本……
您可以使用this question中的一种方法将索引复制到第三维,然后使用该索引简单地索引矩阵.例如(适应Divakar’s solution):
vals = 1:size(mat, 3);
clens = cumsum([3 2 1 1 5 4 2 2 1]);
index = zeros(1, clens(end));
index([1 clens(1:end-1)+1]) = diff([0 vals]);
newMat = mat(:, :, cumsum(index));
然后,您可以将其概括为一个函数,以在多个维度上操作,例如repelem:
function A = my_repelem(A, varargin)
index = cell(1, nargin-1);
for iDim = 1:nargin-1
lens = varargin{iDim};
if isscalar(lens)
if (lens == 1)
index{iDim} = ':';
continue
else
lens = repmat(lens, 1, size(A, iDim));
end
end
vals = 1:size(A, iDim);
clens = cumsum(lens);
index{iDim} = zeros(1, clens(end));
index{iDim}([1 clens(1:end-1)+1]) = diff([0 vals]);
index{iDim} = cumsum(index{iDim});
end
A = A(index{:});
end
对于您的示例数据,您可以像这样使用它:
>> A(:,:,1) = [1 2; 1 2];
>> A(:,:,2) = [2 3; 2 3];
>> A = my_repelem(A, 1, 1, [1 2])
A(:,:,1) =
1 2
1 2
A(:,:,2) =
2 3
2 3
A(:,:,3) =
2 3
2 3