我有以下代码,试图在其中形成数组“ opt”.在这里,我采用了三个可能的值’pos_set’= [1、2、3],并且可以类似的方式对此进行扩展.但是,我只想为pos_set的任何可能的整数值提供通用代码.

    opt = []
    if pos_set == 1:
        for j in range(1, n):
            opt.append([j])
    elif pos_set == 2:
        for j in range(1, n):
            for k in range(j+1, n):
                opt.append([j, k])
    elif pos_set == 3:
        for j in range(1, n):
            for k in range(j+1, n):
                for l in range(k+1, n):
                    opt.append([j, k, l])

为了更清楚起见,我这样做的目的是在滚动n面模具时收集所有可能性,并在保持更大值的情况下继续这样做.

例如,如果您滚动一个序列1-2-6-4,在这种情况下,在较大编号之后获得4. 6,停止滚动.同样,如果您滚动一个序列1-2-6-6,在这种情况下,您会得到一个重复的6,所以您停止了,因为它不大于您之前的滚动.我正在考虑在两个案例中出现较小或相同数字之前的案例,即[1、2、6].

如果你们能帮助我,我将不胜感激.

解决方法:

您可以使用以下递归函数：

def f(p, n, i=1):
    if p == 0:
        return [[]]
    return [[j, *l] for j in range(i, n) for l in f(p - 1, n, j + 1)]

以便：

print(f(1, 7))
print(f(2, 7))
print(f(3, 7))

输出：

[[1], [2], [3], [4], [5], [6]]
[[1, 2], [1, 3], [1, 4], [1, 5], [1, 6], [2, 3], [2, 4], [2, 5], [2, 6], [3, 4], [3, 5], [3, 6], [4, 5], [4, 6], [5, 6]]
[[1, 2, 3], [1, 2, 4], [1, 2, 5], [1, 2, 6], [1, 3, 4], [1, 3, 5], [1, 3, 6], [1, 4, 5], [1, 4, 6], [1, 5, 6], [2, 3, 4], [2, 3, 5], [2, 3, 6], [2, 4, 5], [2, 4, 6], [2, 5, 6], [3, 4, 5], [3, 4, 6], [3, 5, 6], [4, 5, 6]]