從矩陣每列獲取非零行

Question

我有一個矩陣A 假設是：

A=[1 0 8; 
   0 0 2; 
   3 0 5; 
   4 8 0; 
   0 5 3;
   6 1 3;
   1 6 5;
   0 7 1] 

我想在新矩陣中獲取每列的非零行下標。
在我的示例中，

B = [ 1 3 4 6 7 0 0 0; 
      4 5 6 7 8 0 0 0; 
      1 2 3 5 6 7 8 0] 

就大小而言，如果A=(m,n) ，則B將為B=(n,m) （轉置）。 就內容而言，如上所述， B包含A非零行的下標。

Answer 1

這是一種方法：

mask = ~sort(~A);     %// destination of row indexes in output
[ii,~]=find(A);       %// get the row indexes
B = zeros(size(A));
B(mask) = ii; B=B.'   %'//write indexes to output and transpose

Answer 2

我認為這可以滿足您的要求（每列獲取非零行索引）。 這與另一個問題非常相似：

[r c] = size(A);
M = bsxfun(@times, A~=0, 1:size(A,2)).'; %'// substitute values by indices
[~, rows] = sort(M~=0,'descend'); %//'' push zeros to the end
cols = repmat(1:r,c,1);
ind = sub2ind([c r],rows(:),cols(:));
B = repmat(NaN,c,r);
B(:) = M(ind).';
B = B.';

結果：

>> B

B =

     1     3     4     6     7     0     0     0
     4     5     6     7     8     0     0     0
     1     2     3     5     6     7     8     0