Python - 读取pickle数据的索引

时间:2017-03-23 03:28:45

标签: python indexing iteration pickle

拥有以下腌制数据:

[array([[[148, 124, 115],
        [150, 127, 116],
        [154, 129, 121],
        ..., 
        [159, 142, 133],
        [159, 142, 133],
        [161, 145, 142]]]), array([1])]

我能够检索datalabel,如下所示:

data = batch[0]
labels = batch[1]

在这种情况下,在为数据和标签分别创建print时,我有以下输出:

[[[148 124 115]
  [150 127 116]
  [154 129 121]
  ...,
  [159 142 133]
  [159 142 133]
  [161 145 142]]]
[1]

当我添加新图像时,我的批处理文件现在看起来如下,我没有弄清楚如何读取第二个图像及其标签。我似乎无法理解这里如何将pickle文件编入索引:

[array([[[148, 124, 115],
        [150, 127, 116],
        [154, 129, 121],
        ..., 
        [159, 142, 133],
        [159, 142, 133],
        [161, 145, 142]],

       [[165, 136, 145],
        [176, 137, 141],
        [178, 138, 144],
        ..., 
        [199, 163, 171],
        [202, 163, 167],
        [200, 158, 163]]]), array([1, 1])]

如何迭代这样的腌制文件?该文件是如何编入索引的?特别是我想添加更多图像及其标签。

感谢。

1 个答案:

答案 0 :(得分:1)

您的数据被腌制的方式,两个图像最终在同一个数组中,因此您必须相应地索引:

batch[0][0] #this will give the first image
batch[0][1] #this will give the second image
batch[1][0] #this will give the first label
batch[1][1] #this will give the second label