python - 从基于python中另一个列表的列表创建子列表

Question

我有两个列表:l1 和 l2，如下所示。

l1 = [0, 1, 1, 0, 1]
l2 = [0, 0, 1, 1, 0]

我想根据 l2 中的值将 l1 子列表放入列表列表中，如下所示，一个用于“0”，一个用于“1”

[[0, 1, 1], [1, 0]]

这基本上是说，对于 l2 中的 0，l1 中有三个值；对于 l2 中的 1，l1 中有两个值。

有没有简单的 python 方法来做到这一点？

编辑:这是我为解决问题所做的工作。这很好用，但看起来笨重且奇怪。我正在寻找更优雅的解决方案。

d = [[],[]]
for i in range(len(l2)):
    d[l2[i]].append(l1[i])

Answer 1

[[l1[n] for n, l2s in enumerate(l2) if l2s == element] for element in set(l2)]

像这样的事情可能会成功。 set 只是快了一点。