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

Question

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

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

Answer 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))