开发者

How to generate a matrix of combinations

开发者 https://www.devze.com 2023-01-20 17:11 出处:网络
I have 5 items each of which can take on the value of 1 or -1. I want to generate a matrix that consists of rows of the possible combinations. The order of the items does not matter and the order of t

I have 5 items each of which can take on the value of 1 or -1. I want to generate a matrix that consists of rows of the possible combinations. The order of the items does not matter and the order of the combinations does not matter. I know I could do this mechanically, but I thought that someone must know a shortcut to generating this matrix. I apologize if this is similar to other questions but none of the solution开发者_StackOverflows I have found can be applied to this particular problem with my programming skills.


expand.grid(c(-1,1), c(-1,1), c(-1,1), c(-1,1), c(-1,1))


To generalize Greg's answer:

N   <- 5
vec <- c(-1, 1)
lst <- lapply(numeric(N), function(x) vec)
as.matrix(expand.grid(lst))


Alternative from data.table package is slightly faster compared to expand.grid:

library(data.table)  
CJ(c(-1,1), c(-1,1), c(-1,1), c(-1,1), c(-1,1))
0

精彩评论

暂无评论...
验证码 换一张
取 消