如何根据另一个列表中的连续元素对一个列表中的项目进行分组？

Question

我有一个简单的数据排序问题：

我有两个列表：

列表 1： [1,1,1,2,3,3,4,5,5]列表 2： [objectA, objectB, objectC, objectD, ... objectI]

列表具有相同的长度。 我想将 List2 排序为嵌套列表，如下所示：

列表 3： [[objectA, objectB, objectC], [objectD], ...]

所以基本上根据 list1 将 list2 排序为子列表。 这是列表问题还是字典问题，我该如何解决？

Answer 1

我认为解决它的最好方法是使用字典根据List1中的元素对List2的元素进行分组

对于你的例子应该是这样的

List1 = [1,1,1,2,3,3,4,5,5]
List2 = ['objectA', 'objectB', 'objectC', 'objectD', ... 'objectI']

grouping = {}

for key, value in zip(List1, List2):
  if key not in grouping:
    grouping[key] = []
  grouping[key].append(value)

List3 = [grouping[key] for key in sorted(grouping.keys())]

print(List3)

Answer 2

你可以使用类似的东西：

from collections import Counter

list1=[1,1,1,2,2,3,4]
list2=["objectA", "objectB", "objectC", "objectD","objectE","objectF","objectG"]
to_dict=dict(Counter(list1).items()) #{1: 3, 2: 2, 3: 1, 4: 1}

final=[]
i=0
for k,v in to_dict.items():
    final.append(list2[i:v+i])
    i+=v

输出（最终） ：

[
     ['objectA', 'objectB', 'objectC'],
     ['objectD', 'objectE'],
     ['objectF'],
     ['objectG']
]