Efficient Way to find index of max column for each row

Hey I've an matrix of size n x m. I want to find the index of max_col corresponding to each row. I know, i can use a loop to check but is there any other efficient way to do that?

thanks in advance.

Answer 1

[Les Beckham]

Are you wanting to find the column number of the maximum value in each row? If so, this is how you can do that:

A = magic(5) % example data

A = 5×5

17 24 1 8 15 23 5 7 14 16 4 6 13 20 22 10 12 19 21 3 11 18 25 2 9

[m, idx] = max(A, [], 2) % max value from each row and column number where the max value occurs

m = 5×1

24 23 22 21 25

idx = 5×1

2 1 5 4 3

If not, please clarify what you are trying to do.

Answer 2

[Keshav]

Hi, Based on my understanding you have a matrix. for each row you want to find the index of maximum value. you can use the below code to do that.

x = [1 2 3; 5 3 4;10 8 9]
[mx,idx] = max(x,[],2)

here idx(i) will represent the index of column with max value in ith row.

Answer 3

[Harshit Gupta]

Hi, I understand that you're trying to find an efficient way to find the maximum values in each row.

You can do this simply with the max() function on the matrix without using a for loop.

Here's an example:

M=[2 4 0 1;12 3 5 2;7 10 4 1;3 5 20 7]
[max_values, idx] = max(M') %row maximums

You can even get the column maximums and the maximum value of the matrix

[max_values, idx] = max(M) %col maximums
[matrix_max, idx] = max(M(:)) %matrix maximum

Hope this helps!

Answer 4

[Anay Aggarwal]

Hi Varsha,

I have an understanding that you want to find the column number of maximum element in each row of a matrix.

You can perform the following operation:

x = [1 2 3; 4 50 6;11 8 9]

x = 3×3

1 2 3 4 50 6 11 8 9

[mx,idx] = max(x,[],2)

mx = 3×1

3 50 11

idx = 3×1

3 2 1

Hope this helps

Regards