python - 映射两个列表而不循环

Question

我有两个等长的列表。第一个列表 l1 包含数据。

l1 = [2, 3, 5, 7, 8, 10, ... , 23]

第二个列表 l2 包含 l1 中的数据所属的类别:

l2 = [1, 1, 2, 1, 3, 4, ... , 3]

我如何根据第二个列表中的 1, 2, 3, 4 等数字定义的位置来划分第一个列表，使用列表理解或lambda 函数。例如，第一个列表中的 2, 3, 7 属于同一分区，因为它们在第二个列表中具有相应的值。

分区数一开始就知道了。

Answer 1

你可以使用字典:

>>> l1 = [2, 3, 5, 7, 8, 10, 23] 
>>> l2 = [1, 1, 2, 1, 3, 4, 3]

>>> d = {}
>>> for i, j in zip(l1, l2):
...     d.setdefault(j, []).append(i)
... 
>>> 
>>> d
{1: [2, 3, 7], 2: [5], 3: [8, 23], 4: [10]}