通过根据列表B分组生成列表A的子列表

Posted 2021-05-07

我目前有：

a = [939, 84, 100, 338, 281, 94, 884, 848]
b = ["red", "green", "orange", 'blue']*2
zip_list = zip(a, b)

哪个返回：

[(939, 'red'), (84, 'green'), (100, 'orange'), (338, 'blue'), (281, 'red'), 
(94, 'green'), (884, 'orange'), (838, 'blue')]

我的目标是从这里获得：

   # blue #   # green # # orange #   # red #
[ [338, 838], [84, 94], [100,884], [939, 281] ]

即从列表a派生的新列表，按列表b分组（注意字母顺序）。

b中唯一元素的长度将始终是a长度，例如len(a) == 8和len(unique(b)) == 4的一个因素（从数学意义上讲）>>

我目前有：a = [939、84、100、338、281、94、884、848] b = [“红色”，“绿色”，“橙色”，“蓝色”] * 2 zip_list = zip（ a，b）返回：[（939，'red'），（84，'green'），（100，'orange'），（338，'...

答案

这是我尝试使用itertools.groupby

另一答案

使用defaultdict是我想要的。您可以在参数中指定defaultdict，以指定密钥的值类型。