按项目对元组列表进行分组

Question

我以这个列表为例：

[(148, Decimal('3.0')), (325, Decimal('3.0')), (148, Decimal('2.0')), (183, Decimal('1.0')), (308, Decimal('1.0')), (530, Decimal('1.0')), (594, Decimal('1.0')), (686, Decimal('1.0')), (756, Decimal('1.0')), (806, Decimal('1.0'))]

现在我想按id分组，所以我将使用itemgetter(0):

import operator, itertools
from decimal import *
test=[(148, Decimal('3.0')), (325, Decimal('3.0')), (148, Decimal('2.0')), (183, Decimal('1.0')), (308, Decimal('1.0')), (530, Decimal('1.0')), (594, Decimal('1.0')), (686, Decimal('1.0')), (756, Decimal('1.0')), (806, Decimal('1.0'))]

for _k, data in itertools.groupby(test, operator.itemgetter(0)):
    print list(data)

我不知道为什么，但我得到了这个错误的输出：

[(148, Decimal('3.0'))]
[(325, Decimal('3.0'))]
[(148, Decimal('2.0'))]
[(183, Decimal('1.0'))]
[(308, Decimal('1.0'))]
[(530, Decimal('1.0'))]
[(594, Decimal('1.0'))]
[(686, Decimal('1.0'))]
[(756, Decimal('1.0'))]
[(806, Decimal('1.0'))]

如您所见，输出未按 id 分组。但是，如果我使用 itemgetter(1)，上面的代码可以正常工作。输出按十进制值分组。

[(148, Decimal('3.0')), (325, Decimal('3.0'))]
[(148, Decimal('2.0'))]
[(183, Decimal('1.0')), (308, Decimal('1.0')), (530, Decimal('1.0')), (594, Decimal('1.0')), (686, Decimal('1.0')), (756, Decimal('1.0')), (806, Decimal('1.0'))]

我在这里缺少什么？

Answer 1

itertools.groupby()要求数据一致或有序

[(148, Decimal('3.0')), (148, Decimal('2.0')), (325, Decimal('3.0'))] 会工作，但 [(148, Decimal('3.0')), (325, Decimal('3.0')), (148, Decimal('2.0'))] 不会，因为 ID 是 148, 325, 148 而不是 148, 148, 325。

Answer 2

您首先需要排序数据，groupby 才能工作，它将连续元素 根据您提供的密钥：

import operator, itertools
from decimal import *
test=[(148, Decimal('3.0')), (325, Decimal('3.0')), (148, Decimal('2.0')), (183, Decimal('1.0')), (308, Decimal('1.0')), (530, Decimal('1.0')), (594, Decimal('1.0')), (686, Decimal('1.0')), (756, Decimal('1.0')), (806, Decimal('1.0'))]

for _k, data in itertools.groupby(sorted(test), operator.itemgetter(0)):
    print list(data)

但是你最好使用字典来分组以避免不必要的 O(n log n) 排序：

from collections import defaultdict

d = defaultdict(list)

for t in test:
    d[t[0]].append(t)

for v in d.values():
    print(v)

两者都会给你相同的分组，只是顺序不一定相同。

按项目对元组列表进行分组

Group list of tuples by item

python

itertools

python-2.7