python - 如何使用 for 循环从 2d numpy 数组动态创建 3d numpy 数组

Question

我想使用 for 循环从 2d numpy 数组创建一个 3d numpy 数组。我尝试了许多不同的方法从 2d 创建 3d 数组，但每次都会出错。这就是我所做的，结束数组的尺寸应为 (10,3,3)。

#this is a sample code
arr=[]
for i in range(10):
    a=np.random.rand(3,3)
    arr=np.stack(a,arr)
    #arr=np.append(arr,a)
    #arr=np.array([arr,a])
    #arr[i]=a

Answer 1

您可以使用 list.append 方法将 2d 数组追加到 arr 列表中，完成 for 循环后，通过用 np.array 包装将 arr 转换为 3d 数组:

arr = []
for i in range(10):
    a = np.random.rand(3,3)
    arr.append(a)

np.array(arr).shape
# (10, 3, 3)

或者numpy.stack:

np.stack(arr).shape
# (10, 3, 3)