view find_elements.m @ 555:8e0067aef5b0

Make spdiag accept both row and column vectors
author Jonatan Werpers <jonatan@werpers.com>
date Tue, 29 Aug 2017 12:22:01 +0200
parents 48b6fb693025
children
line wrap: on
line source

% I = find_elements(a,b)
% Finds the index of elements a in b.
% a and b have to be in the same order.
function I = find_elements(a,b)
    I = [];

    j = 1;
    for i = 1:length(a)
        while b(j) ~= a(i)
            j = j + 1;
        end
        I(end+1) = j;
        j = j+1;
    end

    assert(length(I) == length(a),'Expected %d but got %d elements',length(a),length(I))
end