组索引向量来自组中元素数量较多的向量

Question

我想从包含每个组中元素数量的向量创建一个包含每个元素的组标识符的向量。

例子：

E = [2 3 4]'

我正在寻找一个向量如下：

I = [1 1 2 2 2 3 3 3 3]

我找到了一个涉及循环的解决方案：

I = [];
for e=1:size(E,1),
    I = [I ; e*ones(E(e),1)];
end

但这看起来不太优雅。欢迎提出任何改进建议。

Answer 1

我也有类似的问题。我认为，如果不使用for循环，这并不像人们想象的那么简单。

以下是我的解决方案：

I = cell2mat(arrayfun(@(x) repmat(x,E(x),1),1:numel(E),'UniformOutput',false)') 

一些解释：x是E的索引。数组函数从1“迭代”到numel（E）。repmat将索引复制x E（x）次。array函数的输出是一个带有向量[11]、[222]和[3333]的1x3单元阵列。由于矢量大小不同，我必须将Uniformoutput设置为false。但我想连接这些嵌套向量，所以我使用cell2mat（…）

Answer 2

试试这个：

X = arrayfun(@(x) [1 zeros(1,x-1)], E, 'uni',0)
Y = cumsum( [X{:}] )

Answer 3

您正在寻找运行长度解码<试试这个

n = sum( E ); %// tot number of elelments
I = zeros( 1, n ); % //preallocate!
I(cumsum( [ 1 E(1:end-1) ] ) ) = 1;
I = cumsum( I )

请参阅ideone的运行示例。