Matlab:每行或每列的第一个非零元素

时间:2013-10-27 23:11:15

标签: matlab matrix sparse-matrix sparse-array

例如,

A = [ -1   0  -2   0   0
   2   8   0   1   0
   0   0   3   0  -2
   0  -3   2   0   0
   1   2   0   0  -4];

如何获得每行的第一个非零元素的向量?

3 个答案:

答案 0 :(得分:16)

您可以使用max

>> [sel, c] = max( A ~=0, [], 2 );

sel等于零的行 - 全部为零,c中的相应列应忽略。

结果:

>> [sel c]= max( A~=0, [], 2 )

sel =
 1
 1
 1
 1
 1
c =
 1
 1
 3
 2
 1

为了找到第一个非零行索引(对于每一列),您只需在第一维上应用max

>> [sel r] = max( A~=0, [], 1 );

答案 1 :(得分:9)

这是一个基于accumarray的解决方案,即使行全为零也能正常工作。

A = [ -1   0  -2   0   0
   2   8   0   1   0
   0   0   3   0  -2
   0  -3   2   0   0
   1   2   0   0  -4];

[r,c] = find(A);

%# for every row, take the minimum column index and put NaN if none is found
firstIndex = accumarray(r,c,[size(A,1),1],@min,NaN);

答案 2 :(得分:1)

您可以通过对每行执行find函数来执行此操作,如下所示:

A = [ -1   0  -2   0   0
   2   8   0   1   0
   0   0   3   0  -2
   0  -3   2   0   0
   1   2   0   0  -4];

% make cell of rows
cellOfRows = num2cell(A, 2);

% apply find function for each row
indexOfFirstNonZeroValues = cellfun(@(row) find(row, 1, 'first'), cellOfRows);


indexOfFirstNonZeroValues =

     1
     1
     3
     2
     1