Получить все возможные комбинации по строкам в матрице
у меня есть 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
как показано здесь и передача каждого элемента этого списка в 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