2015-07-06 12 views
5

Powiedzmy mam listę wskaźników, takich jak:R: Lista indeksów do macierzy binarnej

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

Które określić niezerowe elementy macierzy, takich jak:

(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 

Co jest najszybszy sposób, używając R, do utworzenia m z l, co oznacza, że ​​macierz jest bardzo duża, powiedzmy 50 000 wierszy i 2000 kolumn?

+0

Skąd wiesz, kiedy należy umieścić, powiedzmy, 5? –

+0

Zgaduję, że wartość "5" jest literówką. – akrun

Odpowiedz

5

Spróbuj

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 

Albo zamiast stack, moglibyśmy użyć melt

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

lub używając tylko 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

Uwielbiam indeksowanie macierzy w trzeciej próbie. Moja ulubiona, mało używana funkcja R! – Aaron

+0

@Aaron Dzięki za komentarze. Myślę, że 'sparseMatrix' będzie szybszy niż indeksowanie. – akrun

+0

Dzięki, wszystkie z nich są pomocne! – Misconstruction