Numpy ND陣列

Question

我有一個大小為[128,128,128,3]的numpy數組。 您可以看到這是1組16組8圖像堆棧，大小為128x128x3。 我想將其重塑為大小[16,128,128,24]。 我的代碼如下：

b_imgs_96_reshape = np.empty([16,128,128,24])
for i in range(0,128):
    b_imgs_96_reshape[i//8,:,:,(i%8)*3:(i%8)*3+3] = b_imgs_96[i,:,:,:]

但是我得到了錯誤，有時會隨機崩潰。 這是錯誤消息：

TypeError：列表索引必須是整數或切片，而不是元組

誰能幫我解決這個問題？ 非常感謝您的幫助。

Answer 1

如果輸入b_imgs_96具有形狀（ b_imgs_96 ）和標量dtype，則

1）您發布的代碼段中沒有可能觸發您報告錯誤的列表

2）除非您在智能手表上執行此操作，否則此處的存儲不應成為問題； 對於64位dtype確實為128x128x128x3x8〜50MB

您可以並且可能應該避免循環，而應該

 b_imgs_96_reshape = np.moveaxis(b_imgs_96.reshape(16, 8, 128, 128, 3), 1, 3).reshape(16, 128, 128, 24)

Answer 2

我無法重現您的錯誤。 運行良好：

b_imgs_96 = np.ones([128,128,128,3])

b_imgs_96_reshape = np.empty([16,128,128,24])
for i in range(0,128):
    b_imgs_96_reshape[i//8,:,:,(i%8)*3:(i%8)*3+3] = b_imgs_96[i,:,:,:]

如果是間歇性的，我可能會懷疑是內存問題。