2015-07-06 87 views
5

說我有索引列表,如:R:指數以二元矩陣列表

l <- list(c(1,2,3), c(1), c(1,5), c(2, 3, 5)) 

其中指定矩陣中的非零元素,如:

(m <- matrix(c(1,1,1,0,0, 1,0,0,0,0, 1,0,0,0,5, 0,1,1,0,1), nrow=4, byrow=TRUE)) 

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

是什麼最快的方式,使用R,從lm,給出矩陣非常大,說50.000行和2000列?

+0

你怎麼知道什麼時候放,說,5? –

+0

我猜測值'5'是一個錯字。 – akrun

回答

5

嘗試

d1 <- stack(setNames(l, seq_along(l))) 
library(Matrix) 
m1 <- sparseMatrix(as.numeric(d1[,2]), d1[,1], x=1) 
as.matrix(m1) 
#  [,1] [,2] [,3] [,4] [,5] 
#[1,] 1 1 1 0 0 
#[2,] 1 0 0 0 0 
#[3,] 1 0 0 0 1 
#[4,] 0 1 1 0 1 

或代替stack,我們可以使用melt

library(reshape2) 
d2 <- melt(l) 
sparseMatrix(d2[,2], d2[,1],x=1) 

或者只使用base R

Un1 <- unlist(l) 
m1 <- matrix(0, nrow=length(l), ncol=max(Un1)) 
m1[cbind(as.numeric(d1$ind), d1$values)] <- 1 
m1 
+1

在第三次嘗試中喜歡矩陣索引。 R最喜歡的小功能! – Aaron

+0

@Aaron感謝您的評論。我認爲'sparseMatrix'會比索引更快。 – akrun

+0

謝謝,所有這些都很有幫助! – Misconstruction