Como faço uma matriz de uma lista de vetores em R?

102

Objetivo: a partir de uma lista de vetores de igual comprimento, crie uma matriz onde cada vetor se torna uma linha.

Exemplo:

> a <- list()
> for (i in 1:10) a[[i]] <- c(i,1:5)
> a
[[1]]
[1] 1 1 2 3 4 5

[[2]]
[1] 2 1 2 3 4 5

[[3]]
[1] 3 1 2 3 4 5

[[4]]
[1] 4 1 2 3 4 5

[[5]]
[1] 5 1 2 3 4 5

[[6]]
[1] 6 1 2 3 4 5

[[7]]
[1] 7 1 2 3 4 5

[[8]]
[1] 8 1 2 3 4 5

[[9]]
[1] 9 1 2 3 4 5

[[10]]
[1] 10  1  2  3  4  5

Eu quero:

      [,1] [,2] [,3] [,4] [,5] [,6]
 [1,]    1    1    2    3    4    5
 [2,]    2    1    2    3    4    5
 [3,]    3    1    2    3    4    5
 [4,]    4    1    2    3    4    5
 [5,]    5    1    2    3    4    5
 [6,]    6    1    2    3    4    5
 [7,]    7    1    2    3    4    5
 [8,]    8    1    2    3    4    5
 [9,]    9    1    2    3    4    5
[10,]   10    1    2    3    4    5 
Christopher DuBois
fonte

Respostas:

124

Uma opção é usar do.call():

 > do.call(rbind, a)
      [,1] [,2] [,3] [,4] [,5] [,6]
 [1,]    1    1    2    3    4    5
 [2,]    2    1    2    3    4    5
 [3,]    3    1    2    3    4    5
 [4,]    4    1    2    3    4    5
 [5,]    5    1    2    3    4    5
 [6,]    6    1    2    3    4    5
 [7,]    7    1    2    3    4    5
 [8,]    8    1    2    3    4    5
 [9,]    9    1    2    3    4    5
[10,]   10    1    2    3    4    5
Christopher DuBois
fonte
5
Portanto, a diferença entre isso e o rbind () padrão é que do.call () passa cada item da lista como um argumento separado - certo? do.call (rbind, a) é equivalente a rbind (a [[1]], a [[2]] ... a [[10]])?
Matt Parker
5
do.call () é ótimo para esse propósito, gostaria que fosse melhor "documentado" nos materiais introdutórios.
andrewj
16

simplify2arrayé uma função básica bastante intuitiva. No entanto, como o padrão de R é preencher os dados por colunas primeiro, você precisará transpor a saída. ( sapplyusa simplify2array, conforme documentado em help(sapply).)

> t(simplify2array(a))
      [,1] [,2] [,3] [,4] [,5] [,6]
 [1,]    1    1    2    3    4    5
 [2,]    2    1    2    3    4    5
 [3,]    3    1    2    3    4    5
 [4,]    4    1    2    3    4    5
 [5,]    5    1    2    3    4    5
 [6,]    6    1    2    3    4    5
 [7,]    7    1    2    3    4    5
 [8,]    8    1    2    3    4    5
 [9,]    9    1    2    3    4    5
[10,]   10    1    2    3    4    5
Kalin
fonte
12

Não é simples, mas funciona:

> t(sapply(a, unlist))
      [,1] [,2] [,3] [,4] [,5] [,6]
 [1,]    1    1    2    3    4    5
 [2,]    2    1    2    3    4    5
 [3,]    3    1    2    3    4    5
 [4,]    4    1    2    3    4    5
 [5,]    5    1    2    3    4    5
 [6,]    6    1    2    3    4    5
 [7,]    7    1    2    3    4    5
 [8,]    8    1    2    3    4    5
 [9,]    9    1    2    3    4    5
[10,]   10    1    2    3    4    5
Paolo
fonte
1
Com rjsonresultados, colMeansfunciona apenas para este método! Obrigado!
mpyw
10

A matrixfunção embutida tem a boa opção de inserir dados byrow. Combine isso com um unlistem sua lista de fontes para obter uma matriz. Também precisamos especificar o número de linhas para que possa quebrar os dados não listados. Isso é:

> matrix(unlist(a), byrow=TRUE, nrow=length(a) )
      [,1] [,2] [,3] [,4] [,5] [,6]
 [1,]    1    1    2    3    4    5
 [2,]    2    1    2    3    4    5
 [3,]    3    1    2    3    4    5
 [4,]    4    1    2    3    4    5
 [5,]    5    1    2    3    4    5
 [6,]    6    1    2    3    4    5
 [7,]    7    1    2    3    4    5
 [8,]    8    1    2    3    4    5
 [9,]    9    1    2    3    4    5
[10,]   10    1    2    3    4    5
Kalin
fonte
Ou preencher uma matriz por colunas e depois transpor: t( matrix( unlist(a), ncol=length(a) ) ).
Kalin
8
t(sapply(a, '[', 1:max(sapply(a, length))))

onde 'a' é uma lista. Funcionaria para tamanhos de linha desiguais

Arihant
fonte
3
> library(plyr)
> as.matrix(ldply(a))
      V1 V2 V3 V4 V5 V6
 [1,]  1  1  2  3  4  5
 [2,]  2  1  2  3  4  5
 [3,]  3  1  2  3  4  5
 [4,]  4  1  2  3  4  5
 [5,]  5  1  2  3  4  5
 [6,]  6  1  2  3  4  5
 [7,]  7  1  2  3  4  5
 [8,]  8  1  2  3  4  5
 [9,]  9  1  2  3  4  5
[10,] 10  1  2  3  4  5
aprender
fonte
1
Isso simplesmente não funcionará se as linhas não tiverem o mesmo comprimento, enquanto do.call (rbind, ...) ainda funciona.
primeiro dia
alguma pista de como fazê-lo funcionar para tamanhos de linhas desiguais com NA para os dados de linhas ausentes?
Arihant
1
@rwst Na verdade, do.call (rbind, ...) não funciona para vetores de comprimento desigual, a menos que você realmente pretenda reutilizar o vetor ao preencher a linha no final. Veja a resposta de Arihant para uma maneira que preenche os NAvalores no final.
Kalin