r - 按矩阵中的行获取所有可能的组合

Question

我有一个 m x n看起来像这样的矩阵:

1 2 3
4 5 6

按行获得所有可能组合的最快方法是什么？在这种情况下，这将是 c(1,4), c(1,5), c(1,6), c(2,4), c(2,5) ... c(3,5), c(3,6)
如何使用矢量化方法解决这个问题？一般是 m x n矩阵将有 n^m这样的组合。

Answer 1

您可以使用 expand.grid函数获取每行元素的所有组合，使用 split 构建行列表如图 here并将该列表的每个元素传递给 expand.grid与 do.call功能:

(m <- rbind(1:3, 4:6))
#      [,1] [,2] [,3]
# [1,]    1    2    3
# [2,]    4    5    6
do.call(expand.grid, split(m, rep(1:nrow(m), ncol(m))))
#   1 2
# 1 1 4
# 2 2 4
# 3 3 4
# 4 1 5
# 5 2 5
# 6 3 5
# 7 1 6
# 8 2 6
# 9 3 6

这是一个使用 3 x 2 矩阵而不是 2 x 3 矩阵的示例:

(m <- matrix(1:6, nrow=3))
#      [,1] [,2]
# [1,]    1    4
# [2,]    2    5
# [3,]    3    6
do.call(expand.grid, split(m, rep(1:nrow(m), ncol(m))))
#   1 2 3
# 1 1 2 3
# 2 4 2 3
# 3 1 5 3
# 4 4 5 3
# 5 1 2 6
# 6 4 2 6
# 7 1 5 6
# 8 4 5 6