Python：重塑3D圖像和標簽的Numpy數組

Question

我正在嘗試將（1397,2）的numpy數組重塑為（1397，50,50,20,1）。該數組有2列

each row  in column1 is of length: 50*50*20  [3D image]
each row in column2 is of  lenght : 2 [labels : [0,1],[1,0]].

基本上是嘗試將3d圖像的第一列解壓縮以饋入CNN中的輸入層。

到目前為止我的代碼

np.array([i[0] for i in data]).reshape(-1,50,50,20,1)

給我一個錯誤：

"ValueError: cannot reshape array of size 1397 into shape (50,50,20,1)"

任何幫助是極大的贊賞

干杯羅賓

Answer 1

好的，因為事實證明列1中的所有行都不具有相同的形狀。 一個是17 * 50 * 50而不是20 * 50 * 50。 現在一切都很好:)

感謝邁克爾！