如何根据 Python 中另一个列表的（子列表）索引对列表进行分区

大佬教程收集整理的这篇文章主要介绍了如何根据 Python 中另一个列表的（子列表）索引对列表进行分区，大佬教程大佬觉得挺不错的，现在分享给大家，也给大家做个参考。

如何解决如何根据 Python 中另一个列表的（子列表）索引对列表进行分区？

我有两个列表，一个包含一些独特的元素（在我的例子中是整数），另一个包含指示元素应该插入到新创建的嵌套列表的哪个子列表中的索引。

elements = [1,2,3,4,5,6]
inDices =  [0,1,1]

expected_result = [[1,2],[3,6],[4,5]]

元素列表仅包含唯一项，可能未排序。 索引列表是“标准化的”，这样较低的索引将始终首先出现。 新的嵌套列表应使用索引来确定元素所属的预期结果的子列表。

我想出了以下函数，但我觉得应该有更简单的方法。

def inDices_to_nested_Lists(inDices: Sequence[int],elements: SequencE):
    result = []
    for i in range(max(inDices)+1):
        subList = []
        for j in range(len(elements)):
            if inDices[j] == i:
                subList.append(elements[j])
        result.append(subList)
    return result

谁能想出一种更简单、可能更像 Python 的方式来实现相同的结果？

解决方法

l = [[] for i in range(max(inDices) + 1)]
for x,y in zip(elements,inDices):
    l[y].append(X)
print(l)

[[1,2],[3,6],[4,5]]