python - 通过指定行长度创建二维锯齿状列表

Question

我有一个整数列表x，我想创建一个由零组成的二维锯齿状列表y，这样行的长度i y 的 code> 由 x 的元素 i 确定。

例如:

x = [1, 3, 4]

然后:

y = [[0], [0, 0, 0,], [0, 0, 0, 0]]

我该怎么做？

Answer 1

试试这个单行:

x = [1, 3, 4]
y = [[0]*s for s in x]

现在 y 将包含预期值:

y
=> [[0], [0, 0, 0], [0, 0, 0, 0]]