2009-09-15 10 views
6

Załóżmy, że mam macierz x, która zawiera 10 wierszy i 2 kolumny. Chcę wygenerować nową macierz M, która zawiera każdą unikalną parę wierszy od x - czyli nową macierz z 55 wierszami i 4 kolumnami.Szybkie generowanie kartezjańskiego produktu macierzy

Np

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 
} 

Więc x jest:

 [,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 

A potem M ma 4 kolumny, dwa pierwsze są jednym rzędzie od x i następny 2 to kolejny wiersz z 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 

Czy istnieje szybszy lub prostszy (lub oba) sposób robienia tego w R?

Odpowiedz

7

Funkcja expand.grid() przydatna dla:

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> 

Teraz masz „n * (n + 1)/2 'podzbiorów i można łatwo indeksować oryginalną macierz.

2

Nie podoba mi się to, co robisz, więc po prostu wyrzucę coś, co może lub nie pomoże.

Oto co myślę o jako iloczyn kartezjański dwóch kolumnach:

expand.grid(x[,1],x[,2]) 
+0

Nigdy nie wiedziałem o expand.grid(). Odpowiedź Dirka przynosi wszystko razem (jak zawsze ...) –

1

Korzystanie odpowiedź Dirka:

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 

Dzięki wszystkim!