在 R 中创建组合的函数

Question

我需要使用最小值为 5 和最大值为 80 的“n”个变量创建组合，它们之间的差异为 x，并且每个组合的值之和应等于 100。

Answer 1

对于n个变量，你需要将可变数量的参数传递给expand.grid。您可以为此使用do.call：

combs = do.call(expand.grid, rep(list(seq(from = 5, to = 80, by = 5)), n))

如果您想自定义名称，您需要单独分配它们（通过setNames）。例如，要获取名称 m1 … mn，请使用：

values = rep(list(seq(from = 5, to = 80, by = 5)), n)
names = paste0('m', seq_len(n))
combs = do.call(expand.grid, setNames(values, names))

此外，我不会将总数存储在同一个 data.frame 中——使用单独的变量：

total = rowSums(combs)

……但这实际上甚至没有必要。

放在一个函数定义中：

my_combs = function (n, from, to, by, total) {
    values = rep(list(seq(from, to, by)), n)
    names = paste0('m', seq_len(n))
    combs = do.call(expand.grid, setNames(values, names))
    combs[rowSums(combs) == total, ]
}

而你（例如）这样称呼它：

result = my_combs(n = 2, from = 5, to = 80, by = 5, total = 100)

产量：

> result
    m1 m2
64  80 20
79  75 25
94  70 30
109 65 35
124 60 40
139 55 45
154 50 50
169 45 55
184 40 60
199 35 65
214 30 70
229 25 75
244 20 80