如何使用列表理解来转换它？

Question

l = []
for i in range(11):
    l.append(1)
    for j in range(i):
        l.append(0)

print(l)

输出遵循一个(1, 1, 0, 1, 0, 0, 1, 0, 0, 0, ...)模式。但是，我不知道如何使用列表理解来转换嵌套 for 循环。

Answer 1

由于需要在每次迭代中添加两个单独的元素，因此很难将其编写为理解式。它可以使用列表来实现sum：

l = sum([[1] + [0] * i for i in range(11)], [])

输出：

[
 1,
 1, 0,
 1, 0, 0,
 1, 0, 0, 0,
 1, 0, 0, 0, 0,
 1, 0, 0, 0, 0, 0,
 1, 0, 0, 0, 0, 0, 0,
 1, 0, 0, 0, 0, 0, 0, 0,
 1, 0, 0, 0, 0, 0, 0, 0, 0,
 1, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
]