2015-07-06 15 views
5

dire che ho un elenco di indici, come:R: Elenco degli indici di matrice binaria

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

che specificano gli elementi diversi da zero in una matrice, come:

(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 

cosa fa il modo più veloce, usando R, per fare m da l, dando che la matrice è molto grande, diciamo 50.000 righe e 2000 colonne?

+0

Come fai a sapere quando mettere, diciamo, 5? –

+0

Immagino che il valore '5' sia un errore di battitura. – akrun

risposta

5

Prova

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 

O invece di stack, potremmo usare melt

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

O utilizzando solo 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

Ama l'indicizzazione della matrice nel terzo tentativo. La mia caratteristica preferita di R! – Aaron

+0

@Aaron Grazie per i commenti. Penso che 'sparseMatrix' sarebbe più veloce dell'indicizzazione. – akrun

+0

Grazie, tutti questi sono utili! – Misconstruction

Problemi correlati