如何使用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)