CODEWITHSUNDEEP
rsortingmatrixindices
N D Thokare
N D Thokare

Reputation: 1743

Get indices of K smallest or largest elements in each row of a matrix in R

How to get indices of K smallest or largest elements in eaach row of a matrix in R?

E.g. I have matrix:

2   3   1  65  2
46  7   9  3   2
9   45  3  5   7
24  65  87 3   6
34  76  54 33  6

I'd like to get Indices matrix of say 2 smallest elements (breaking ties in any way) in each row. the result should be in following format:

3 1
5 4
3 4
4 5
5 4

I tried few commands using sort, apply, arrayInd, which etc. But still unable to get desired result. Any help is welcome.

Upvotes: 8

Views: 7363

Answers (2)

d_n_k_f_m_nn
d_n_k_f_m_nn

Reputation: 1

What about

  • finding the indices of k largest values in each row

    apply(mat, 1, function(x, k) which(x <= max(sort(x, decreasing = F)[1:k]), arr.ind = T), k)`

  • finding the indices of k smallest values in each row

    apply(mat, 1, function(x, k) which(x >= min(sort(x, decreasing = T)[1:k]), arr.ind = T), k)`

On your example, for k <- 2, the former results in

     [,1] [,2] [,3] [,4] [,5]
[1,]    2    1    1    2    2
[2,]    4    3    2    3    3

and the latter results in

[[1]]
[1] 1 3 5

[[2]]
[1] 4 5

[[3]]
[1] 3 4

[[4]]
[1] 4 5

[[5]]
[1] 4 5

Change apply's second parameter from 1 to 2 for searching the columns.

Upvotes: 0

IRTFM
IRTFM

Reputation: 263481

apply(mat, 1, which.max)  #.....largest
apply(mat, 1, which.min)  #.....smallest

t(apply(mat, 1, sort)[ 1:2, ])  # 2 smallest in each row

t(apply(mat, 1, order)[ 1:2, ])  # indices of 2 smallest in each row

Besides using decreasing=TRUE, you could also have used this for the two largest in a row:

t(apply(mat, 1, order)[ 5:4, ])

Upvotes: 13

Related Questions