arrays - 保留每列的 n 个最大值，并在 MATLAB 中将其余值设置为零，无需循环

Question

我有一个未排序数字矩阵，我想保留每列的 n 个最大(不一定唯一)值，并将其余值设置为零。

我想出了如何用循环来完成它:

a = [4 8 12 5; 9 2 6 18; 11 3 9 7; 8 9 12 4]
k = 2

for  n = 1:4
    [y, ind] = sort(a(:,n), 'descend');
    a(ind(k+1:end),n) = 0;
end
a

这给了我:一个=

 4     8    12     5
 9     2     6    18
11     3     9     7
 8     9    12     4

k =

 2

一个=

 0     8    12     0
 9     0     0    18
11     0     0     7
 0     9    12     0

然而，当我试图消除循环时，我似乎无法获得正确的索引，因为:

a = [4 8 12 5; 9 2 6 18; 11 3 9 7; 8 9 12 4]
k = 2

[y, ind] = sort(a, 'descend');
b = ind(k+1:end,:)
a(b) = 0

这给了我这个:(这不是我想做的)一个=

 4     8    12     5
 9     2     6    18
11     3     9     7
 8     9    12     4

k =

 2

b =

 4     3     3     1
 1     2     2     4

一个=

 0     8    12     5
 0     2     6    18
 0     3     9     7
 0     9    12     4

我索引错了吗？我必须使用循环吗？

我引用了这个问题来开始，但这并不是我想要做的:How to find n largest elements in an array and make the other elements zero in matlab?

Answer 1

你非常接近。 sort 函数中的 ind 为您提供每列的行位置，该特定值将出现在已排序的输出中。如果您想正确索引矩阵并消除条目，则需要做一些额外的工作。您知道对于 I 的每一列，这告诉您我们需要从该特定列中删除这些条目。因此，我要做的是使用 I 的每一列作为我们需要消除的行来生成列优先线性索引。

尝试这样做:

a = [4 8 12 5; 9 2 6 18; 11 3 9 7; 8 9 12 4];
k = 2;
[y, ind] = sort(a, 'descend');

%// Change here
b = sub2ind(size(a), ind(k+1:end,:), repmat(1:size(a,2), size(a,1)-k, 1));
a(b) = 0;

我们使用 sub2ind帮助我们生成列主索引，其中行由第 k 元素之后的 ind 中的值表示，我们需要的列是该矩阵中的每一列。在排序后截断 k 值后，还剩下 size(a,1)-k 行，因此我们生成从 1 到尽可能多的列值a 中的列和剩余的行一样多。

我们得到这个输出:

>> a

a =

     0     8    12     0
     9     0     0    18
    11     0     0     7
     0     9    12     0