快速的方式来获得每个矩阵列

Question

TOPN元件这是从recommenderlab包的代码段，该取矩阵的评分，并返回每个用户顶部5个元素 -

reclist <- apply(ratings, MARGIN=2, FUN=function(x) 
    head(order(x, decreasing=TRUE, na.last=NA), 5)) 

对于大型矩阵（> 10K列）运行时间太长，是否有任何方法可以重新编写它以提高效率？也许通过使用dpyr或data.table包）？编写C++代码是不适合我

Answer 1

与data.table和基础R

# 10000 column dummy matrix 
cols <- 10000 
mat <- matrix(rnorm(100*cols), ncol=cols) 

的回答随着data.table一个选项：

library(data.table) 
dt1 <- data.table(mat) 
# sort every column, return first 5 rows 
dt1[, lapply(.SD, sort, decreasing=T)][1:5] 
system.time(dt1[, lapply(.SD, sort, decreasing=T)][1:5]) 

结果：

user system elapsed 
2.904 0.013 2.916 

在普通的老基地，它实际上更快！ （感谢您的评论阿伦）

system.time(head(apply(mat, 2, sort, decreasing=T), 5)) 

user system elapsed 
0.473 0.002 0.475 

然而，无论是比上面的代码示例更快，根据system.time（）

system.time(
apply(mat, MARGIN=2, FUN=function(x) { 
    head(order(x, decreasing=TRUE, na.last=NA), 5) 
})) 

user system elapsed 
3.063 0.031 3.094