如何在Python中根据图像标签来分割图像列表

Question

我有一个名为随机数据的图像列表，它们被标记为57个类。所以，这个列表有两个维度。第一个维度是图像，第二个维度是它们类的标签。

randomData[0]是图像，randomData1是标签。问题是，在Python中，如何使用相同的标签保存不同组中的所有图像？例如，我想要一个标签为0的图像数据集，标签为1的图像数据库，…，标签为9的图像数据集。 我正在尝试编写以下代码：

[randomData[0] for i in randonData[1][:] if i==0]

但是，我不知道如何考虑那些与标签相关的随机数据[0]'。

我很感激你的帮助。

Answer 1

使用字典：

from collections import defaultdict

images = defaultdict(list)

for image, label in randomData:
    images[label].append(image)

for k,v in images.iteritems():
   print('Label {} has {} images'.format(k, len(v)))