提问者:小点点

如何调整numpy数组的大小?


我有NumPy数组x_train,它的形状是(880,):

x_train.shape
(880,)

该数组的每个元素都是具有形状(256,192,3)的图像

x_train[0].shape
(256, 192, 3)

如何重塑x_train数组以获得大小为(880,256,192,3)的数组


共1个答案

匿名用户

如果您试图创建一个numpy数组,并得到了该形状,这意味着并非所有元素都具有相同的形状。 您可以这样验证:

for ix, item in enumerate(xtrain):
    assert item.shape == (256, 192, 3), f'found problematic item at index {ix}'

然后删除有问题的行。