Создание векторизованного массива из списка начальных / конечных индексов
У меня двухколонная матрица 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 ответа
Вот векторизованное решение, которое я хотел бы использовать для этой конкретной задачи, используя функцию 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
cell2mat(arrayfun(@colon,M(:,1)',M(:,2)','UniformOutput',false))
У меня нет IMFILL, но на моей машине этот метод работает быстрее, чем другие предложения, и я думаю, что он побьет метод IMFILL из-за использования find.
Это может быть сделано даже быстрее, если M настроен транспонированным (и мы настраиваем третий и четвертый аргументы arrayfun).
Возможно, есть даже лучшее решение, которое я почему-то не вижу, но вот версия, использующая 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
Очень странное решение ИМХО - создание временных строк и использование EVAL. Может быть также однострочным.
tmp = cellstr(strcat(num2str(M(:,1)),{':'},num2str(M(:,2)),{' '}));
v = eval(['[' cell2mat(tmp') ']']);
Я знаю, что это, вероятно, не будет работать на большой матрице. Просто для удовольствия.