The most efficient way to locate the first 1's in each row of a matrix

Question

Ulrik William Nash 2019년 10월 19일

0
링크

이 질문에 대한 바로 가기 링크

https://kr.mathworks.com/matlabcentral/answers/486328-the-most-efficient-way-to-locate-the-first-1-s-in-each-row-of-a-matrix

댓글: Jos (10584) 2019년 10월 21일

I have a potentially large r by c matrix, m, consisting of 1's and 0's. What is the most efficient way create a r by 1 vector, v, containing the index of the first 1 in each row of m.

For example,

m = [0 1 1;1 0 0; 0 0 1]

should result in

v = [2;1;3]

댓글 수: 0
이전 댓글 -2개 표시이전 댓글 -2개 숨기기

댓글을 달려면 로그인하십시오.

이 질문에 답변하려면 로그인하십시오.

Answer 1

Star Strider 2019년 10월 19일

1
링크

이 답변에 대한 바로 가기 링크

https://kr.mathworks.com/matlabcentral/answers/486328-the-most-efficient-way-to-locate-the-first-1-s-in-each-row-of-a-matrix#answer_397218

MATLAB Online에서 열기

The max function will return the indices of the first instances of the maximum value it encounters, in the desired dimension. Since your matrix only has (0,1), it is appropriate here.

This:

m = [0 1 1;1 0 0; 0 0 1];
[~,v] = max(m,[],2)

produces:

댓글 수: 1
이전 댓글 -1개 표시이전 댓글 -1개 숨기기

Jos (10584) 2019년 10월 21일

MATLAB Online에서 열기

nice! One can shorten it to:

[~, v] = max(m')

... but what if there is no 1 in a row

댓글을 달려면 로그인하십시오.

Answer 2

Jos (10584) 2019년 10월 19일

0
링크

이 답변에 대한 바로 가기 링크

https://kr.mathworks.com/matlabcentral/answers/486328-the-most-efficient-way-to-locate-the-first-1-s-in-each-row-of-a-matrix#answer_397220

MATLAB Online에서 열기

m = [0 1 1 1 ;1 0 1 0; 0 0 1 1 ; 0 0 0 1]
% a simple solution
[r, c] = find(m') ;
clear v1
v1(r,1) = c
% another solution
[v2, ~] = find(cumsum(cumsum(m,2),2)==1)