gerar rapidamente o produto cartesiano de uma matriz
Pergunta
Vamos dizer que tenho um x
matriz que contém 10 linhas e 2 colunas. Eu quero gerar um novo M
matriz que contém cada par exclusivo de linhas de x
-. Isto é, uma nova matriz com 55 linhas e 4 colunas
por exemplo.,
x <- matrix (nrow=10, ncol=2, 1:20)
M <- data.frame(matrix(ncol=4, nrow=55))
k <- 1
for (i in 1:nrow(x))
for (j in i:nrow(x))
{
M[k,] <- unlist(cbind (x[i,], x[j,]))
k <- k + 1
}
Assim, x
é:
[,1] [,2]
[1,] 1 11
[2,] 2 12
[3,] 3 13
[4,] 4 14
[5,] 5 15
[6,] 6 16
[7,] 7 17
[8,] 8 18
[9,] 9 19
[10,] 10 20
E então M
tem 4 colunas, os dois primeiros são uma linha de x
eo próximo 2 são outra linha de x
:
> head(M,10)
X1 X2 X3 X4
1 1 11 1 11
2 1 11 2 12
3 1 11 3 13
4 1 11 4 14
5 1 11 5 15
6 1 11 6 16
7 1 11 7 17
8 1 11 8 18
9 1 11 9 19
10 1 11 10 20
Existe tanto um mais simples (ou ambos) maneira mais rápida ou de fazer isso em R?
Solução
A função expand.grid()
útil para isso:
R> GG <- expand.grid(1:10,1:10)
R> GG <- GG[GG[,1]>=GG[,2],] # trim it to your 55 pairs
R> dim(GG)
[1] 55 2
R> head(GG)
Var1 Var2
1 1 1
2 2 1
3 3 1
4 4 1
5 5 1
6 6 1
R>
Agora você tem a 'n * (n + 1) / 2' subconjuntos e você pode índice simples a sua matriz original.
Outras dicas
Eu não estou bastante grokking que você está fazendo, então eu vou jogar fora algo que pode ou não ajuda.
Aqui está o que eu penso como o produto cartesiano das duas colunas:
expand.grid(x[,1],x[,2])
Você também pode tentar o pacote "relações". Aqui está a vinheta. Ele deve funcionar como este :
relation_table(x %><% x)
Usando resposta de Dirk:
idx <- expand.grid(1:nrow(x), 1:nrow(x))
idx<-idx[idx[,1] >= idx[,2],]
N <- cbind(x[idx[,2],], x[idx[,1],])
> all(M == N)
[1] TRUE
Obrigado a todos!
Inspirado de outras respostas, aqui é uma função executando o produto cartesiano de duas matrizes, no caso de duas matrizes, o produto cartesiano completo, para apenas um argumento, omitindo um de cada par:
cartesian_prod <- function(M1, M2) {
if(missing(M2)) { M2 <- M1
ind <- expand.grid(1:NROW(M1), 1:NROW(M2))
ind <- ind[ind[,1] >= ind[,2],] } else {
ind <- expand.grid(1:NROW(M1), 1:NROW(M2))}
rbind(cbind(M1[ind[,1],], M2[ind[,2],]))
}