根据计数按元素进行数组复制

[poster]

我的问题是与此类似一个 ，但我想根据在相同尺寸的第二阵列指定的计数复制每个元素。

例如，假设我有一个数组v = [3 1 9 4] ，我想使用rep = [2 3 1 5]将第一个元素复制2次，第二个元素复制3次，依此类推以获得[3 3 1 1 1 9 4 4 4 4 4] 。

到目前为止，我正在使用一个简单的循环来完成工作。这是我开始的：

vv = []; for i=1:numel(v) vv = [vv repmat(v(i),1,rep(i))]; end 我设法通过预先分配空间来进行改进：

vv = zeros(1,sum(rep)); c = cumsum([1 rep]); for i=1:numel(v) vv(c(i):c(i)+rep(i)-1) = repmat(v(i),1,rep(i)); end 但是我仍然觉得必须有一种更聪明的方式来做到这一点...谢谢



回答：

这是我喜欢完成此操作的一种方法：

>> index = zeros(1,sum(rep)); >> index(cumsum([1 rep(1:end-1)])) = 1; index = 1 0 1 0 0 1 1 0 0 0 0 >> index = cumsum(index) index = 1 1 2 2 2 3 4 4 4 4 4 >> vv = v(index) vv = 3 3 1 1 1 9 4 4 4 4 4 这是通过首先创建一个索引向量零来实现的，该索引向量的长度与所有值的最终计数相同。通过执行rep向量的累加总和，删除最后一个元素，并在开始处放置1，我得到了一个索引向量， index显示了复制值组的开始位置。这些点标有1。当对index进行累加总和时，我得到了最终的索引向量，可以将其用于索引v以创建异构复制值的向量。



更多&回答...