R矩陣：在最低值的列中輸入零

Question

我有一個包含數值的矩陣nxn。我需要找到哪些是每列中具有最高值的3列和其餘列中的輸入零。

例如，在列將是：

  a b r t  y n 
user1 1.3 2 4 5.5 0 0 
user2  8 0 1.5 5  3 7.2 

所得矩陣將是：

  a b r t  y n 
user1  0 2 4 5.5 0 0 
user2  8 0 0 5  0 7.2 

Answer 1

我們通過使用apply的matrix的行（與MARGIN=1）rank元素環，創建邏輯向量（< 4），使用ifelse來替換不是最高的元素3，並轉置（t）輸出。

m2 <- m1 
m2[] <-t(apply(m1, 1, function(x) 
     ifelse(rank(-x, ties.method='min') < 4, x, 0))) 
m2 
#  a b r t y n 
#user1 0 2 4 5.5 0 0.0 
#user2 8 0 0 5.0 0 7.2 

---

或者另一種選擇是

library(dplyr) 
m1[ave(m1, row(m1), FUN=min_rank) < 4] <- 0 

Answer 2

如果效率的問題，我會嘗試的matrixStats包

m[matrixStats::rowRanks(m) < 4] <- 0 
m 
#  a b r t y n 
# user1 0 2 4 5.5 0 0.0 
# user2 8 0 0 5.0 0 7.2 

如果關係很重要，使用ties.method參數