将元组列表转换为字典
我有一个包含元组的列表,像这样:
[
('a', 1),
('a', 2),
('a', 3),
('b', 1),
('b', 2),
('c', 1),
]
我想要遍历这个列表,并根据每个元组的第一个元素来进行处理。比如,我想打印出类似这样的内容:
a 1 2 3
b 1 2
c 1
我该怎么做才能在循环遍历这些元组的时候,不用额外的变量来跟踪第一个元素是否相同呢?这样感觉有点麻烦(而且我还得先对列表进行排序)……
5 个回答
11
这是一个使用 groupby 的解决方案
from itertools import groupby
l = [('a',1), ('a', 2),('a', 3),('b', 1),('b', 2),('c', 1),]
[(label, [v for l,v in value]) for (label, value) in groupby(l, lambda x:x[0])]
输出结果:
[('a', [1, 2, 3]), ('b', [1, 2]), ('c', [1])]
groupby(l, lambda x:x[0])
会给你一个迭代器,这个迭代器里面包含了
['a', [('a', 1), ...], c, [('c', 1)], ...]
39
稍微简单一点...
from collections import defaultdict
fq = defaultdict(list)
for n, v in myList:
fq[n].append(v)
print(fq) # defaultdict(<type 'list'>, {'a': [1, 2, 3], 'c': [1], 'b': [1, 2]})
48
l = [
('a', 1),
('a', 2),
('a', 3),
('b', 1),
('b', 2),
('c', 1),
]
d = {}
for x, y in l:
d.setdefault(x, []).append(y)
print d
生成的结果是:
{'a': [1, 2, 3], 'c': [1], 'b': [1, 2]}