如何枚舉R中矩陣中的所有組合？

Question

我正在嘗試構建一個包含所有可能組合的矩陣。 例如，

a=(1:2)^3 #=c(1,8)
b=(1:3)^2 #=c(1,4,9)

我想定義c使得c=c(1+1,1+4,1+9,8+1,8+4,8+9) 。 我從上一個問題中學到了如何從函數outer獲得這樣的c 。 我當前的問題是，如何獲得矩陣M ，如下所示：

提前致謝！

Answer 1

我們可以在outer使用expand.grid

data.frame(expand.grid(a, b), c = c(outer(a, b, "+")))

#  Var1 Var2  c
#1    1    1  2
#2    8    1  9
#3    1    4  5
#4    8    4 12
#5    1    9 10
#6    8    9 17

哪里

outer(a, b, "+") #gives

#     [,1] [,2] [,3]
#[1,]    2    5   10
#[2,]    9   12   17

Answer 2

好的，這是：

z <- outer(b, a, "+")
cbind(a[col(z)], b[row(z)], c(z))
#     [,1] [,2] [,3]
#[1,]    1    1    2
#[2,]    1    4    5
#[3,]    1    9   10
#[4,]    8    1    9
#[5,]    8    4   12
#[6,]    8    9   17

---

一種稍微適應的expand.grid解決方案。

ref <- expand.grid(b = b, a = a)
val <- do.call("+", ref)  ## or `rowSums(ref)` with an implicit `as.matrix`
cbind(ref, c = val)
#  b a  c
#1 1 1  2
#2 4 1  5
#3 9 1 10
#4 1 8  9
#5 4 8 12
#6 9 8 17

在這種情況下，結果是數據幀而不是矩陣。

Answer 3

或另一種選擇是CJ

library(data.table)
CJ(a, b)[, C := V1 + V2][]
#.  V1 V2  C
#1:  1  1  2
#2:  1  4  5
#3:  1  9 10
#4:  8  1  9
#5:  8  4 12
#6:  8  9 17