Question

我有一个双列矩阵M，它包含一堆间隔的开始/结束索引：

startInd   EndInd
1          3
6          10
12         12
15         16

如何生成所有区间索引的向量：

v = [1 2 3 6 7 8 9 10 12 15 16];

我正在使用循环进行上述操作，但我想知道是否有更优雅的矢量化解决方案？

v = [];
for i=1:size(M,1)
    v = [v M(i,1):M(i,2)];
end

Answer 1

这是我想用这个特定问题的矢量化解决方案，使用函数cumsum：

v = zeros(1, max(endInd)+1);  % An array of zeroes
v(startInd) = 1;              % Place 1 at the starts of the intervals
v(endInd+1) = v(endInd+1)-1;  % Add -1 one index after the ends of the intervals
v = find(cumsum(v));          % Perform a cumulative sum and find the nonzero entries

Answer 2

cell2mat(arrayfun(@colon,M(:,1)',M(:,2)','UniformOutput',false))

我没有IMFILL，但是在我的机器上，这种方法比其他建议更快，我认为由于使用了find，它会超过IMFILL方法。

如果将M设置为转置（并且我们调整arrayfun的第三个和第四个参数），它可以更快。

Answer 3

可能有一个更好的解决方案，我不知道怎么看，但这是一个使用IMFILL的版本

startInd = [1,6,12,15];
endInd = [3,10,12,16];

%# create a logical vector with starts and ends set to true to prepare for imfill
tf = false(endInd(end),1);
tf([startInd,endInd]) = true;

%# fill at startInd+1 wherever startInd is not equal endInd
tf = imfill(tf,startInd(startInd~=endInd)'+1); %' SO formatting

%# use find to get the indices
v = find(tf)'  %' SO formatting

v =
     1     2     3     6     7     8     9    10    12    15    16

Answer 4

非常奇怪的解决方案恕我直言创建临时字符串和使用EVAL。也可以是单行。

tmp = cellstr(strcat(num2str(M(:,1)),{':'},num2str(M(:,2)),{' '}));
v = eval(['[' cell2mat(tmp') ']']);

我知道它可能不适用于大型矩阵。只是为了好玩。

从开始/结束索引列表创建矢量化数组

4 个答案: