将相同字典键的值放在python中的一个组中

时间:2017-09-19 23:43:48

标签: python list dictionary iteration key-value

我有两个清单:

list1=[0,0,0,1,1,2,2,3,3,4,4,5,5,5]
list2=['a','b','c','d','e','f','k','o','n','q','t','z','w','l']

dictionary=dict(zip(list1,list2))

我想为每个键输出相同键的值,如下所示:

 0 ['a','b','c']  
 1 ['d','e']  
 2 ['f','k']  
 3 ['o','n']  
 4 ['q','t']  
 5 ['z','w','l'] 

我编写了以下代码来执行此操作,但它并不像我想的那样工作

for k,v in dictionary.items():  

     print (k,v)

你能告诉我如何修改代码,以便我能达到预期的结果吗? 提前谢谢!

3 个答案:

答案 0 :(得分:3)

您展示的代码与您所描述的内容不同。

除此之外,您可以通过先压缩列表然后使用collections.defaultdict对同一个键的值进行分组来将相同键的值组合在一起:

from collections import defaultdict

d = defaultdict(list)
for k, v in zip(list1, list2):
    d[k].append(v)
print(d)
# defaultdict(<type 'list'>, {0: ['a', 'b', 'c'], 1: ['d', 'e'], 2: ['f', 'k'], 3: ['o', 'n'], 4: ['q', 't'], 5: ['z', 'w', 'l']})

答案 1 :(得分:3)

关于您的代码:

dictionary = dict(zip(list1, list2))

创建字典:

{0: 'c', 1: 'e', 2: 'k', 3: 'n', 4: 't', 5: 'l'}

除了每组中的最后一个值之外都会丢失。您需要处理压缩列表以构建分组数据。有itertools.groupby()defaultdict(list)两种方式,如下所示。

使用collections.defaultdict个列表对包含list1键和list2值的项目进行分组。将每个列表中的项目与zip()配对:

from collections import defaultdict

list1=[0,0,0,1,1,2,2,3,3,4,4,5,5,5]
list2=['a','b','c','d','e','f','k','o','n','q','t','z','w','l']

d = defaultdict(list)

for k,v in zip(list1, list2):
    d[k].append(v)

for k in sorted(d):
    print('{} {!r}'.format(k, d[k]))

输出:

0 ['a', 'b', 'c']
1 ['d', 'e']
2 ['f', 'k']
3 ['o', 'n']
4 ['q', 't']
5 ['z', 'w', 'l']

由于字典中的项目是无序的,因此输出按键排序。

答案 2 :(得分:3)

您可以使用itertool.groupby获得简洁的单行解决方案:

import itertools
list1=[0,0,0,1,1,2,2,3,3,4,4,5,5,5]
list2=['a','b','c','d','e','f','k','o','n','q','t','z','w','l']
final_list = {a:[i[-1] for i in list(b)] for a, b in itertools.groupby(zip(list1, list2), key=lambda x: x[0])}
for a, b in final_list.items():
    print(a, b)

输出:

0 ['a', 'b', 'c']
1 ['d', 'e']
2 ['f', 'k']
3 ['o', 'n']
4 ['q', 't']
5 ['z', 'w', 'l']