1

我有一个元组列表。第一部分是一个标识符，可以重复也可以不重复。我想将此列表处理成字典，由标识符键入。问题是，我一直无法考虑按键覆盖：

def response_items(self):
        ri = self.response_items_listing#(gets the list)          
        response_items = {}
        for k, g in groupby(ri, itemgetter(0)):
            x = list(g)
            l = [(xx[1],xx[2]) for xx in x]
            response_items[k] = l
        return response_items

例如，像这样的列表：

[('123', 'abc', 'def'),('123', 'efg', 'hij'),('456', 'klm','nop')]

会回来

{123:('efg', 'hij'), 456:('klm', 'nop')}

但是我需要：

{123:[('abc', 'def'),('efg', 'hij')], 456:('klm', 'nop')}

我需要采取一个步骤来按键合并/聚合，但我没有完全看到它。

什么是更好或更有效的解决方案？

4

3 回答 3

2

一个简单的方法是

from collections import defaultdict

ri = [('123', 'abc', 'def'),('123', 'efg', 'hij'),('456', 'klm','nop')]
response_items = defaultdict(list)
for r in ri:
    response_items[r[0]].append(r[1:])
print response_items

这使

defaultdict(<type 'list'>, {'123': [('abc', 'def'), ('efg', 'hij')],
                            '456': [('klm', 'nop')]})

如果你想

defaultdict(<type 'list'>, {'123': ['abc', 'def', 'efg', 'hij'],
                            '456': ['klm', 'nop']})

作为输出，您可以使用response_items[r[0]].extend(r[1:]).

于 2012-10-22T16:23:44.823 回答
2

你可以使用setdefault()：

In [79]: dic={}
In [80]: for x in lis:
    dic.setdefault(x[0],[]).append(x[1:])
   ....:     
   ....:     

In [82]: dic
Out[82]: {'123': [('abc', 'def'), ('efg', 'hij')], '456': [('klm', 'nop')]}
于 2012-10-22T16:20:43.340 回答
1

If there's some reason to be using itertools.groupby, then you can avoid using a defaultdict or setdefault methodology - [in fact, if you want to go down those routes, then you don't really need the groupby!] - by:

mydict = {}
for k, g in groupby(some_list, itemgetter(0)):
    mydict[k] = [el[1:] for el in g]
于 2012-10-22T16:27:42.207 回答