I have a 2D array in a format of (image, ID) , where some images have the same ID. For example:
data_2d = [(image1, 1) , (image2, 2),(image3, 2) , image(4, 2) , (image5,3), (image6,3)]
Now my goal is to convert this to a 3D array using their id numbers. To be more clear:
data_3d = [[(image1, 1) ] , [(image2, 2),(image3, 2) , image(4, 2)] , [(image5,3), (image6,3)]]
Is there a way to this?
Use a dictionary (or a defaultdict) to group the tuples by id, and then convert the dictionary's values to a list:
from collections import defaultdict
data_2d = [('image1', 1), ('image2', 2), ('image3', 2), ('image4', 2), ('image5', 3), ('image6', 3)]
groups = defaultdict(list)
for e in data_2d:
groups[e[1]].append(e)
data_3d = list(groups.values())
print(data_3d)
Output
[[('image1', 1)], [('image2', 2), ('image3', 2), ('image4', 2)], [('image5', 3), ('image6', 3)]]
Note that I change your variable names to strings to make the code runable.
