创建可变大小数组

Question

我想创建一个如下所示的 numpy 数组，但我希望能够创建一个可变形状数组。因此，对于下面的一个，它将是 n = 3。有没有一种巧妙的方法可以用 numpy 做到这一点，或者我需要一个 for 循环。

输出数据：

import numpy as np

np.array([1,0,0],[0,1,0],[0,0,1],[1,1,1],[0,0,0])

Answer 1

假设您要创建名称为d 的数组，其行数为row，列数为col。 它还将用 0 初始化数组的所有元素。

d = [[0 for x in range(col)] for y in range(row)]

您可以通过 d[i][j] 访问任何元素 I,j。