将字典转换为多维数组？

Question

我有存储 4 个类的彩色图像数据的字典。 每个图像是 256* 256 * 3。我的字典有 4 个键，分别命名为飞机、鸟、狗和猫。 字典中的每个键都有 50 个维度为 256* 256* 3 的数组（每个类的 50 张图像作为 3-D 数组，总共 200 张图像）。 我想将此数据结构转换为 200 * 3* 256* 256 形状数组，该数组已从字典键中提取标签。 我怎样才能实现这种最简单的方法？ 我尝试了 numpy reshape 但没有奏效。

Answer 1

所以我假设你的数据设置看起来像这样（显然是真实数据而不是随机数据）：

classes = ["plane", "bird", "dog", "cat"]
images = {
     k: [np.random.uniform(size=(256, 256, 3))
         for _ in range(50)]
     for k in classes
}

我们可以做的是：

X = np.concatenate([images[k] for k in classes], axis=0)
y = np.concatenate([[i] * len(images[k]) for i, k in enumerate(classes)])

这给了你想要的，我相信：

>>> X.shape
(200, 256, 256, 3)

>>> y.shape
(200,)