在矩阵中按行获取所有可能的组合
Posted
技术标签:
【中文标题】在矩阵中按行获取所有可能的组合【英文标题】:Get all possible combinations by row in matrix 【发布时间】:2015-11-14 20:04:33 【问题描述】:我有一个 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
这样的组合。
【问题讨论】:
【参考方案1】:您可以使用expand.grid
函数获取每行中元素的所有组合,使用split
构建行列表,如here 所示,并使用@ 将该列表的每个元素传递给expand.grid
987654325@函数:
(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
【讨论】:
expand.grid(split(m, row(m)))
可能更简单以上是关于在矩阵中按行获取所有可能的组合的主要内容,如果未能解决你的问题,请参考以下文章