我觉得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']}
有没有更好的办法?
最佳答案
原文链接:https://www.f2er.com/python/439497.html