在 R 中使用 combn 创建所有可能组合的矩阵

Question

这与我正在做的一道作业题有关。我需要对一些向量进行数据操作到矩阵中，助教建议使用 combn 函数：

# what I'm starting with
a = c(1, 2)
b = c(NA, 4, 5)
c = c(7, 8)

# what I need to get 
my_matrix
a    b    c
1   NA    7
1   NA    8
1    4    7 
1    4    8 
1    5    7 
1    5    8 
2   NA    7 
2   NA    8 
2    4    7 
2    4    8
2    5    7
2    5    8

my_matrix 是一个矩阵，其中包含 a、b 和 c 中元素的所有可能组合，列名称为 a、b 和 c。我明白 combn() 在做什么，但不确定如何将其转换为上面显示的矩阵？

在此先感谢您的帮助！

Answer 1

expand.grid，在问题的评论中提到，是更好、更容易的方法。但是你也可以使用 combn

#STEP 1: Get all combinations of elements of 'a', 'b', and 'c' taken 3 at a time
temp = t(combn(c(a, b, c), 3))

# STEP 2: In the first column, only keep values present in 'a'
#Repeat STEP 2 for second column with 'b', third column with 'c'
#Use setNames to rename the column names as you want
ans = setNames(data.frame(temp[temp[,1] %in% a & temp[,2] %in% b & temp[,3] %in% c,]),
                                                                  nm = c('a','b','c'))
ans
#   a  b c
#1  1 NA 7
#2  1 NA 8
#3  1  4 7
#4  1  4 8
#5  1  5 7
#6  1  5 8
#7  2 NA 7
#8  2 NA 8
#9  2  4 7
#10 2  4 8
#11 2  5 7
#12 2  5 8

在 R 中使用 combn 创建所有可能组合的矩阵

Using combn in R to create a matrix of all possible combinations

r

data-manipulation