如何在 numpy 数组中保存多个带有标签的图像？

【问题标题】：How can I save multiple images with labels in a numpy array?如何在 numpy 数组中保存多个带有标签的图像？
【发布时间】：2018-07-06 14:37:10
【问题描述】：

以前有人问过这类问题。但我的有点不同。例如，我在一个文件夹中有 20 个文件（20 个矩阵），每个矩阵的尺寸为 40*40。此外，这 20 个文件代表 20 个不同的类别。

现在我想创建一个 Numpy 数组，其中长度为 20*2。在每一行的第一列，我想存储一个 40*40 的矩阵。因此，我想要涵盖的所有 20 个文件。然后我想将标签保留为数字，例如1,2,3 .... 20. 稍后，当我调用 createdNumpyArray[0,0] 时，它应该显示第一个矩阵。如何在 Python 中做到这一点？

【问题讨论】：

标签： python numpy multidimensional-array numpy-ndarray

【解决方案1】：

我会使用字典来存储从键到图像的映射。

ind_to_image = {
    0: numpy array with 40 x 40 shape, 
    ...,
    19: numpy array with 40 x 40 shape,
}

并将索引（键）保存到 20 x 2 数组的第一列因为我认为您不能直接将 numpy 数组保存到另一个数组的元素中。下面的代码会得到一个ValueError

arr1 = np.zeros((20, 2))
arr1[0, 0] = np.zeros((40, 40))

【讨论】：