创建可变大小的数组

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。