的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。它的一切都很好:)

感谢迈克尔！