python - 根据现有数组创建二维数组

Question

我有一个数组为 [2, 3, 1, 3, 1]
我想根据 [2, 3, 1, 3, 1] 定义一个全为 0 的二维数组，它将类似于

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

如何在没有硬编码的情况下做到这一点？

标签： pythonarrays

Answer 1

一个简单的理解就可以了：

array = [2, 3, 1, 3, 1]

zeroes = [[0] * x for x in array]
# [[0, 0], [0, 0, 0], [0], [0, 0, 0], [0]]