itertools.groupby（）

Question

我有这样的数据：

self.data = list: [(1, 1, 5.0), 
        (1, 2, 3.0), 
        (1, 3, 4.0), 
        (2, 1, 4.0), 
        (2, 2, 2.0), 
        (2, 3, 4.0), 
        (2, 5, 3.0), 
        (3, 2, 2.0), 
        (3, 4, 4.0), 
        (3, 5, 3.0)] 

当我运行这段代码：

for mid, group in itertools.groupby(self.data, key=operator.itemgetter(0)): 

为list(group)我得到：

list: [(1, 1, 5.0), 
     (1, 2, 3.0), 
     (1, 3, 4.0)] 

这就是我想要的。

但是，如果使用1而不是0

for mid, group in itertools.groupby(self.data, key=operator.itemgetter(1)): 

到组由元组的第二个数字，我只得到：

list: [(1, 1, 5.0)] 

即使是具有“1等的元组“在那1（2）的位置。

Answer 1

itertools.groupby收集在一起连续具有相同键的项目。 如果您希望所有项目具有相同的密钥，则必须先排序self.data。

for mid, group in itertools.groupby(
    sorted(self.data,key=operator.itemgetter(1)), key=operator.itemgetter(1)): 

Answer 2

没有排序的变体（通过字典）。应该更好地表现明智。

def full_group_by(l, key=lambda x: x): 
    d = defaultdict(list) 
    for item in l: 
     d[key(item)].append(item) 
    return d.items()