Matlab: сортировка 2D-матрицы и сохранение узлов в треугольных группах - PullRequest
Новая
       16

Matlab: сортировка 2D-матрицы и сохранение узлов в треугольных группах

0 голосов
/ 30 апреля 2019

Я пытаюсь оптимизировать скрипт Matlab (ниже), который находит ограничивающие рамки для функциональных значений всех нижних левых треугольников в 2D-пространстве. Код проходит через все треугольники, а затем сортирует узлы в порядке возрастания на основе функциональных значений. Это кажется неэффективным.

Есть ли способ, которым я мог бы отсортировать функциональные значения перед циклами, но все же сохранить узлы в группах треугольников? Или какой-то другой умный способ ускорить процесс? 

clear;

x = (1:600)';
y = (1:500);
z = 2 * x.^2 + y;

zGrid = linspace(min(z, [], 'all') - 1, max(z, [], 'all') + 1, 200);

for iX = 1:length(x) - 1
    for iY = 1:length(y) - 1
        % Node indices
        xIndices = [iX, iX, iX + 1];
        yIndices = [iY, iY + 1, iY];

        % Node values
        xTmp = x(xIndices);
        yTmp = y(yIndices);
        zTmp = z(sub2ind(size(z), xIndices, yIndices));

        % Node sorted according to z
        [zSorted, indicesSorted] = sort(zTmp);
        xSorted = xTmp(indicesSorted);
        ySorted = yTmp(indicesSorted);

        % Get bounding box on zGrid
        iMin = find(zGrid <= zSorted(1), 1, 'last');
        iMax = find(zGrid(iMin:end) >= zSorted(end), 1, 'first') + (iMin - 1);
    end
end

1 Ответ

0 голосов
/ 30 апреля 2019

Вы можете использовать meshgrid для генерации всех индексов, а затем просто адаптировать код для получения желаемого результата: 

x = (1:600).';
y = (1:500);
z = 2 * x.^2 + y;

zGrid = linspace(min(z(:)) - 1, max(z(:)) + 1, 200);
[X,Y] = meshgrid(x(1:end-1),y(1:end-1));
X = X(:);
Y = Y(:);

% Node indices
xIndices = [X, X, X + 1];
yIndices = [Y, Y + 1, Y];

% Node values
xTmp = x(xIndices);
yTmp = y(yIndices);
zTmp = z(sub2ind(size(z), xIndices, yIndices));

% Node sorted according to z
[zSorted, indicesSorted] = sort(zTmp,2);
xSorted = xTmp(indicesSorted+[0:3:(length(X)-1)*3].');
ySorted = yTmp(indicesSorted+[0:3:(length(Y)-1)*3].');

% Get bounding box on zGrid
% I use fliplr to get the last value and not the first one
[~,iMin] = max(fliplr(zGrid <= zSorted(:,1)),[], 2);
iMin = 200-iMin+1; %first to last value conversion
[~,iMax] = max(zGrid(iMin:end) >= zSorted(:,end),[],2);
iMax = iMax + iMin -1;
...