MATLAB: Replace value with index in 2D array

Question

Hi I have a 2D array like this

 A=[0 0 1; 1 0 1; 0 1 0]

I want to replace 1 in each row with column index value. e.g new matrix will be like this:

result=[0 0 3 ; 1 0 3 ; 0 2 0]

Thanks in advance

Answer 1

This works:

A=[0 0 1; 1 0 1; 0 1 0];
[~,CIV] = find(A);                      % ‘CIV’ = ‘Column Index Value’
A(A>0) = CIV
result = A
result =
       0     0     3
       1     0     3
       0     2     0