2015-07-06 21 views
5

sagen, dass ich eine Liste von Indizes, wie:R: Liste der Indizes binären Matrix

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

, die die Nicht-Null-Elemente in einer Matrix angeben, wie:

(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 

Was ist der schnellste Weg, mit R, m von l machen, so dass die Matrix ist sehr groß, sagen 50.000 Reihen und 2000 Spalten?

+0

Wie Sie wissen, wann zu setzen, sagen wir, 5 verwenden könnte? –

+0

Ich rate den Wert "5" ist ein Tippfehler. – akrun

Antwort

5

Versuchen

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 

Oder statt stack wir melt

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

Oder mit nur 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

Liebe die Matrix Indizierung in Ihrem dritten Versuch. Meine Lieblings-wenig genutzte Funktion von R! – Aaron

+0

@Aaron Danke für die Kommentare. Ich denke, die 'SparseMatrix' wäre schneller als die Indizierung. – akrun

+0

Danke, alle diese sind hilfreich! – Misconstruction