关于python：可变数量的相关嵌套循环


variable number of dependent nested loops

给出两个整数n和d，我想构造一个长度为d的所有非负元组的列表，这些元组的总和为n，包括所有排列。这类似于整数分区问题，但是解决方案要简单得多。例如
d==3：

1
2
3
4
5
[
    [n-i-j, j, i]
    for i in range(n+1)
    for j in range(n-i+1)
]

可以很容易地扩展到更多维度，例如d==5：

1
2
3
4
5
6
7
[
    [n-i-j-k-l, l, k, j, i]
    for i in range(n+1)
    for j in range(n-i+1)
    for k in range(n-i-j+1)
    for l in range(n-i-j-l+1)
]

我现在想制作d，即嵌套循环的数量，一个变量，但是我不确定当时如何嵌套循环。

有任何提示吗？

相关讨论

递归：首先创建一个长度为d-1的元组列表，该元组遍历所有ijk，然后用另一列n-sum(ijk)完整该列表。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
def partition(n, d, depth=0):
    if d == depth:
        return [[]]
    return [
        item + [i]
        for i in range(n+1)
        for item in partition(n-i, d, depth=depth+1)
        ]


# extend with n-sum(entries)
n = 5
d = 3
lst = [[n-sum(p)] + p for p in partition(n, d-1)]

print(lst)
相关讨论