将具有多维值的一维数组转换为多维数组

Question

我正在处理深度学习中的图像，并且我有一个由此声明的多维 Python 数组：

train_images = temp1.apply(format_image).values

当我执行此代码时：

print(train_images[0])

我得到了 RGB 值的 3 维数组。

当我执行此代码时：

print(train_images[0].shape)

它给出 (224, 224, 3)

当我执行此操作时：

print(train_images[0].shape)

我得到 (1547,)

我相信我需要它的大小（1547,224,224,3）才能使机器学习算法正常工作。 我以前肯定遇到过，但不记得我是如何解决的，有什么想法吗？

Answer 1

如果您所有的图像都具有相同的形状，您可以使用reshape

import numpy as np

train_images = np.reshape(train_images, (train_images[0], 224, 224, 3))