arrays - 在一个衬里中选择数组中的 n 个最大值

Question

没有特别的原因，我想知道是否有可能在一个数组或矩阵中选择 n 个最大值，所有这些都在 oneliner 中。

假设我有数组:

A=randn(1,100);

我想要最大的值，例如 20 个值，最好从高到低排序。到目前为止，我有 2-liner:

A_ordered = sort(A,'descend');
A_big20 = A_ordered(1:20);

有谁知道这是否可以在一行中实现？如果是这样，我将不胜感激线索、想法或答案!

Answer 1

你说你经常这样做，所以最简单和最好的方法是在 MATLAB 路径上定义一个函数:

function B = topN(A, n)

    if nargin==2 && n==0 
        B = []; return; end

    B = sort(A(:));
    if nargin>1 && n < numel(A)           
        B = B(1:n); end

end

并调用你的“一个”类轮:

top_ten = topN( randn(1,100), 10 );

如果您希望它们按原始顺序排列(而不是排序):

function B = topN_unsorted(A, n)

    if nargin>1 && n==0 
        B = []; return; end

    if nargin==1 || n > numel(A)
        B = A; 
    else
        [B,I] = sort(A(:));
        B = B(I(1:n));
    end

end