Obtenir les indices des n plus grands éléments dans une matrice

Supposons que j'ai la matrice suivante:

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

Et je veux les indices dans le top 5 des éléments (dans ce cas, 32, 13, 12, 11, 10). Quelle est la façon la plus propre de faire cela dans MATLAB?

  • Une précision: Comment voudriez-vous traiter avec des éléments répétés? Par exemple, si le numéro 32 est apparu 7 fois, vous souhaitez obtenir des indices pour l'ensemble des 7, ou seulement cinq d'entre eux, ou seulement 1 d'entre eux et les indices pour les 4 plus grands éléments?
  • Leschinski s'il vous Plaît ne pas ajouter des balises de titres, il n'est pas nécessaire et généralement découragés par la communauté (voir cette méta post pour la réponse officielle sur ce sujet)
InformationsquelleAutor Vlad the Impala | 2010-04-22
  1. 77

    Il existe quelques façons dont vous pouvez faire ce en fonction de comment vous voulez traiter avec des valeurs répétées. Voici une solution qui trouve des indices pour les 5 valeurs les plus grandes (qui pourraient inclure des valeurs répétées) à l'aide de 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

    Voici une solution qui trouve les 5 plus gros unique valeurs, puis trouve tous éléments égaux à ceux des valeurs (qui pourrait être de plus de 5 si il y a des valeurs répétées), à l'aide de unique et 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
    • En fait, il n'y a pas besoin de faire la dernière étape posés par gnovice. La deuxième sortie de trier une liste de balises pour le tri. De sorte que l'indice de ces valeurs sélectionnées est donné par les balises de la sorte. Si vous souhaitez de l'indice en termes d'une paire d'indices, puis d'appeler ind2sub.
    • Bonne prise. J'étais en train de changer, comme vous avez dit, ainsi que l'ajout d'une autre option pour traiter des valeurs répétées d'une façon différente.
    • Cette méthode est probablement la plus propre pour les petites matrices, mais clairement sous-optimale pour les grandes matrices puisque le tri des échelles comme O(N*log(N)) tout en faisant le "non Matlab façon" sera mise à l'échelle comme O(N).
    • UNIQUE retourne déjà triés valeurs et dans un format vectoriel, donc, juste unique(A) dans la première ligne devrait fonctionner.
    • Bonne prise. Fixe.
    • Merci pour ce très concise et précise réponse à cette question. Vous m'a beaucoup aidée!

    InformationsquelleAutor gnovice
  2. 16

    Si vous avez un grand tableau et que vous souhaitez uniquement quelques éléments en dehors de ça. Ce serait ma solution. 

    Arraycopy = Array;
for j = 1:n
   [a, Index(j)] = max(Arraycopy);
   Arraycopy(Index(j)) = -inf;
end
maximumValues = Array(Index);

    Je pense qu'il devrait être plus rapide et moins de RAM exigeant que le type de solution.

    • Cela fonctionne seulement si les n plus grands éléments sont supérieures à 0. Sinon, remplacer le 0 par min(Arraycopy, ...).
    InformationsquelleAutor Sebastian
  3. 7

    Vous pouvez trouver de bonnes réponses à matlab questions aussi sur matlabcentral. J'ai trouvé une bonne mex mise en œuvre, tandis que la recherche de la même chose.

    Il est fait par Bruno Luong à l'aide d'un partiel de tri rapide de l'algorithme mis en œuvre avec le C-MEX. La complexité est O(n + k.log(k)), où n est la taille du tableau, et k est le nombre d'éléments sélectionnés. Il est plus rapide que le TRI ou plusieurs appels de MIN/MAX pour la grande taille des entrées. Multidimensionnelle de la capacité de prises en charge

    http://www.mathworks.com/matlabcentral/fileexchange/23576-minmax-selection

    InformationsquelleAutor seviyor
  4. 3

    Dans MATLAB ≥ R2017b, vous pouvez utiliser maxk pour ce but précis.

    [maxvalues, ind] = maxk(A(:), 5);
    • Merci de remarquer que l'un, je n'avais pas encore vu!
    • Nice! On dirait qu'il a le même comportement que la première solution dans ma réponse (c'est à dire la sortie peut inclure des valeurs répétées).
    InformationsquelleAutor Sardar Usama