python - 基于共同元素的拆分列表？

Question

我有大量包含数据的元组；

test = [
    ('admin', 1, 2),
    ('admin', 3, 4),
    ('admin', 5, 6),
    ('user', 1, 2),
    ('user', 3, 4),
    ('user', 5, 6),
    ('guest', 1, 2),
    ('guest', 3, 4),
    ('guest', 5, 6)
   ]

如何根据元组的 element0 将它们分成 3 个单独的迭代？例如;

[('**admin**', 1, 2),('admin', 3, 4),('admin', 5, 6)]

[('**user**', 1, 2),('user', 3, 4),('user', 5, 6),]

[('**guest**', 1, 2),('guest', 3, 4),('guest', 5, 6)]

提前致谢！布雷特

Answer 1

from collections import defaultdict

buckets = defaultdict(list)
for tup in test:
    buckets[tup[0]].append(tup)

Answer 2

itertools.groupby与列表理解一起使用：

>>> test = [('admin', 1, 2), ('admin', 3, 4), ('admin', 5, 6), ('user', 1, 2), ('user', 3, 4), ('user', 5, 6), ('guest', 1, 2), ('guest', 3, 4), ('guest', 5, 6)]
>>>
>>> from operator import itemgetter
>>> from itertools import groupby
>>> 
>>> [list(g) for k, g in groupby(sorted(test, key=itemgetter(0)), itemgetter(0))]
[[('admin', 1, 2), ('admin', 3, 4), ('admin', 5, 6)], [('user', 1, 2), ('user', 3, 4), ('user', 5, 6)], [('guest', 1, 2), ('guest', 3, 4), ('guest', 5, 6)]]

Answer 3

iterables = {}
for x in test:
    iterables.setdefault(x[0], []).append(x)

然后你的迭代是iterables.values().