создание векторизованного массива из списка начальных / конечных индексов - PullRequest
11 голосов
/ 11 мая 2010

У меня есть матрица из двух столбцов 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

Ответы [ 4 ]

9 голосов
/ 11 мая 2010

Вот векторизованное решение, которое я хотел бы использовать для этой конкретной задачи, используя функцию 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
6 голосов
/ 31 января 2011
cell2mat(arrayfun(@colon,M(:,1)',M(:,2)','UniformOutput',false))

У меня нет IMFILL, но на моей машине этот метод работает быстрее, чем другие предложения, и я думаю, что он побьет метод IMFILL из-за использования find.

Это может быть сделано даже быстрее, если M настроен транспонированным (и мы настраиваем третий и четвертый аргументы arrayfun).

1 голос
/ 11 мая 2010

Возможно, есть даже лучшее решение, которое я почему-то не вижу, но вот версия, использующая 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
0 голосов
/ 11 мая 2010

Очень странное решение, ИМХО создание временных строк и использование EVAL.Может быть также однострочным.

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

Я знаю, что, вероятно, не будет работать на большой матрице.Просто для удовольствия.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...