I'm trying to do what the following MATLAB code does in python but I can't quite understand the output of it. This is the MATLAB code
A = zeros((48,60));
A(3,5) = 1;
A(3,6) = 1;
A(48,60) = 1;
A(47,60) = 1;
A(24,30) = 1;
A(38,45) = 1;
f = find(A == 1);The last line with f returns [195;243;1416;2150;2879;2880] which is a column of numbers. But I don't get how these numbers are obtained from find(). It seems like it isn't multiplying the row and column numbers. Why am I getting those numbers? Also, what should I do to get the same thing in python?
A= np.zeros(48, 60)
A[2, 4] = 1
A[2, 5] = 1
A[47, 59] = 1
A[46, 59] = 1
A[23, 29] = 1
A[37, 44] = 1
np.where(A ==1)I've tried np.where(A== 1) but it only returns indices of the rows and columns.
1 Answer
The find command in matlab returns the linear indices of each element. More can be find here: .
The equivalent python code would then be np.argwhere which also returns the indices:
To get the exact same behavior as matlab, you have to transform the matrix into a vector:
A= np.zeros((48, 60))
A[2, 4] = 1
A[2, 5] = 1
A[47, 59] = 1
A[46, 59] = 1
A[23, 29] = 1
A[37, 44] = 1
np.argwhere(np.ravel(A)==1)
