使用可变数量的 for 循环创建元组

Question

给定n 和k，我需要创建所有长度为k 的元组，其条目来自range(n)（0 到 n-1），这样元组的条目按字典顺序排列并且有括号以特定格式。具体来说，元组从内到外在每一对周围都有括号。

例如，如果n=3 和k=4，那么我希望输出包含(((0,0),1),2) 之类的内容，但不 包含(((0,0),2),1) 之类的内容。

下面的代码适用于这个特定的实例。问题是我不知道如何概括k，这是下面代码中的for 循环数。我只能为特定的k 执行此操作，例如此处的k=4。我真的需要能够为k 的任何值做到这一点。

n=3
k=4
my_list = []
for a in range(n):
    x = a
    for b in range(a,n):
        y = (x,b)
        for c in range(b,n):
            z = (y,c)
            for d in range(c,n):
                w = (z,d)
                my_list.append(w)
print my_list

输出：

[(((0, 0), 0), 0), (((0, 0), 0), 1), (((0, 0), 0), 2), (((0, 0), 1), 1), (((0, 0), 1), 2), (((0, 0), 2), 2), (((0, 1), 1), 1), (((0, 1), 1), 2), (((0, 1), 2), 2), (((0, 2), 2), 2), (((1, 1), 1), 1), (((1, 1), 1), 2), (((1, 1), 2), 2), (((1, 2), 2), 2), (((2, 2), 2), 2)]

Answer 1

另一种解决方案，基于itertools.combinations_with_replacement：

import itertools

n, k = 4, 4
f = lambda x: (f(x[:-1]), x[-1]) if len(x)>2 else x

print(*map(f, itertools.combinations_with_replacement(range(n), k)), sep='\n')

打印：

(((0, 0), 0), 0)
(((0, 0), 0), 1)
(((0, 0), 0), 2)
(((0, 0), 0), 3)
(((0, 0), 1), 1)
...and so on.

Answer 2

以下应该可以工作。

def tuples(n, k):
    if k == 2:
        return [(i,j) for i in range(n) for j in range(i, n)]
    else:
        return [(t, m) for t in tuples(n, k-1) for m in range(t[1], n)]

基本情况会在结果中精确生成您需要的元组。

从递归情况产生的每个元组(x,y)，我们创建一组元组((x, y), z)和z >= y。

Answer 3

试试这个，我希望它是相关的：

n=4
k=4
my_list=[(((a,b),c),d) for a in range(n) for b in range(a,n) for c in range(b,n) for d in range(c,n) ]
print my_list

输出：

[(((0, 0), 0), 0), (((0, 0), 0), 1), (((0, 0), 0), 2), (((0 , 0), 0), 3), (((0, 0), 1), 1), (((0, 0), 1), 2), (((0, 0), 1), 3), (((0, 0), 2), 2), (((0, 0), 2), 3), (((0, 0), 3), 3), (((0, 1), 1), 1), (( (0, 1), 1), 2), (((0, 1), 1), 3), (((0, 1), 2), 2), (((0, 1), 2), 3) , (((0, 1), 3), 3), (((0, 2), 2), 2), (((0, 2), 2), 3), (((0, 2), 3), 3), (((0, 3), 3), 3), (((1, 1), 1), 1), (((1, 1), 1), 2), (((1, 1), 1), 3), (((1, 1), 2), 2), (((1, 1), 2), 3), (((1, 1), 3), 3), (( (1, 2), 2), 2), (((1, 2), 2), 3), (((1, 2), 3), 3), (((1, 3), 3), 3) , (((2, 2), 2), 2), (((2, 2), 2), 3), (((2, 2), 3), 3), (((2, 3), 3), 3), (((3, 3), 3), 3)]

Answer 4

你见过 itertools 内置库吗？ https://docs.python.org/3.1/library/itertools.html?highlight=combinations#itertools.combinations_with_replacement

乍一看您的问题，看起来您实际上是在创建一个包含替换的所有组合的列表（尽管您真的想使用 k+1）。这不会有“嵌套”属性，但它会给你准确的数字，我想调整结构比提出你需要的递归算法要容易得多。这是一个例子：

import itertools

n = 3
k = 4
choices = range(n)
output = list(itertools.combinations_with_replacement(choices, k + 1)
# output = [(0, 0, 0, 0), (0, 0, 0, 1), ...]