如何找到至少包含k个非零元素的列的索引？

Question

我的矩阵A如下：

A =

     1     2     0     0     4
     0     0     0     1     3
     0     0     1     1     1

我想找到A列的索引，其中每列具有不止k个nonero元素。 我用过：

find(all(A~=0));

，但是它返回存在至少一个非零元素的列的索引。

Answer 1

如果逐步进行，可能会有所帮助。
首先，找到每列中非零元素的数量：

nNonZero = sum(A~=0)

然后找到大于k的列

find(nNonZero>k)

Answer 2

与其检查行中所有 all 的值都不为零，而是计算有多少个值，然后应用阈值：

k = 3
find(sum(A~=0,2)>=k)

返回：

ans =

   1
   3