Question

我有一定的功能：

def create4（n）：

output = []
for i in range(n):
    output.append(range(1+i,1))
return output

我希望它能产生：

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

当我打印create4（5）时。如何更改for循环使代码生成所需的输出？

Answer 1

添加第二个for循环。第一个for将迭代每个嵌套的list。第二个for将添加元素。

def create4(n):
    output = []
    for i in range(n):
        output.append([])
        for j in range(i, n):
            output[i] += [j + 1]
    return output

print create4(5)

使用for循环创建具有较低列表的多个列表

1 个答案: