Get the indices of the n largest elements in a matrix

前端 未结 4 2054
情书的邮戳
情书的邮戳 2020-11-27 15:46

Suppose I have the following matrix:

01 02 03 06
03 05 07 02
13 10 11 12
32 01 08 03

And I want the indices of the top 5 elements (in this

4条回答
  •  半阙折子戏
    2020-11-27 16:39

    There are a couple ways you can do this depending on how you want to deal with repeated values. Here's a solution that finds indices for the 5 largest values (which could include repeated values) using sort:

    [~, sortIndex] = sort(A(:), 'descend');  % Sort the values in descending order
    maxIndex = sortIndex(1:5);  % Get a linear index into A of the 5 largest values
    

    Here's a solution that finds the 5 largest unique values, then finds all elements equal to those values (which could be more than 5 if there are repeated values), using unique and ismember:

    sortedValues = unique(A(:));          % Unique sorted values
    maxValues = sortedValues(end-4:end);  % Get the 5 largest values
    maxIndex = ismember(A, maxValues);    % Get a logical index of all values
                                          %   equal to the 5 largest values
    

提交回复
热议问题