按键分组项目？

Question

我觉得Python应该有一个内置的功能.获取项目列表并将其转换为字典,将键映射到具有该键共同的项目列表.

这很容易做到:

# using defaultdict
lookup = collections.defaultdict(list)
for item in items:
    lookup[key(item)].append(item)

# or, using plain dict
lookup = {}
for item in items:
    lookup.setdefault(key(item), []).append(item)

但这是一个常见的用例,内置函数会很好.我可以自己实现它,如下:

def grouped(iterable, key):
    result = {}
    for item in iterable:
        result.setdefault(key(item), []).append(item)
    return result

lookup = grouped(items, key)

这与itertools.groupby几个重要方面不同.要得到相同的结果groupby,你必须这样做,这有点难看:

lookup = dict((k, list(v)) for k, v in groupby(sorted(items, key=key), key))

一些例子:

>>> items = range(10)
>>> grouped(items, lambda x: x % 2)
{0: [0, 2, 4, 6, 8], 1: [1, 3, 5, 7, 9]}

>>> items = 'hello stack overflow how are you'.split()
>>> grouped(items, len)
{8: ['overflow'], 3: ['how', 'are', 'you'], 5: ['hello', 'stack']}

有没有更好的办法？

Answer 1

我还将这个问题发布到 comp.lang.python 上，共识似乎是这实际上并不常见，不足以保证内置函数。因此，使用显而易见的方法是最好的。它们可以工作并且可读。

# using defaultdict
lookup = collections.defaultdict(list)
for item in items:
    lookup[key(item)].append(item)

# or, using plain dict
lookup = {}
for item in items:
    lookup.setdefault(key(item), []).append(item)

我本来打算删除我的问题，但我不妨将其留在这里，以防有人偶然发现它来寻找信息。