如何将多个列表映射到一个字典?

13

我使用了这段代码:

dictionary = dict(zip(list1, list2))

为了将两个列表映射到一个字典中。其中:

list1 = ('a','b','c')
list2 = ('1','2','3')

该词典等于:

{'a': 1, 'c': 3, 'b': 2}

有办法添加第三个列表吗:

list3 = ('4','5','6')

这样词典就会等于:

{'a': [1,4], 'c': [3,5], 'b': [2,6]}

需要添加第三个列表,以便它跟随现有的映射。

想法是在for循环中迭代执行此操作,并将多个值正确地映射到关键字。这种做法是否可行?


请注意,您的输入是元组而不是列表,在这种情况下它们基本相同。列表使用方括号,元组使用圆括号。 - wjandrea
'c': [3,5], 'b': [2,6] -- Aren't those numbers in the wrong order? Shouldn't they be 'c': [3, 6], 'b': [2, 5]? Also FWIW, dicts are unordered, so if it's easier to read, I'm proposing: {'a': [1, 4], 'b': [2, 5], 'c': [3, 6]} - wjandrea
2个回答

19
dict((z[0], list(z[1:])) for z in zip(list1, list2, list3))

这将可以正常工作。或者,如果您喜欢更优美的字典推导式语法:

{z[0]: list(z[1:]) for z in zip(list1, list2, list3)}

这可以轻松扩展到任意数量的列表:

list_of_lists = [list1, list2, list3, ...]
{z[0]: list(z[1:]) for z in zip(*list_of_lists)} 

如果您想将类型转换以确保值列表仅包含整数:

def to_int(iterable):
    return [int(x) for x in iterable]

{z[0]: to_int(z[1:]) for z in zip(*list_of_lists)}

当然,你可以在一行代码里完成这个任务,但我宁愿不这样做。


在我看来,使用可迭代解包最容易阅读:{k: list(t) for k, *t in zip(list1, list2, list3)}。然后嵌套推导式也不错:{k: [int(x) for x in t] for k, *t in zip(*list_of_lists)}。或者你可以使用 map:{k: list(map(int, t)) for k, *t in zip(*list_of_lists)} - wjandrea

12
In [12]: list1 = ('a','b','c')

In [13]: list2 = ('1','2','3')

In [14]: list3 = ('4','5','6')

In [15]: zip(list2, list3)
Out[15]: [('1', '4'), ('2', '5'), ('3', '6')]

In [16]: dict(zip(list1, zip(list2, list3)))
Out[16]: {'a': ('1', '4'), 'b': ('2', '5'), 'c': ('3', '6')}

In [17]: dict(zip(list1, zip(map(int, list2), map(int, list3))))
Out[17]: {'a': (1, 4), 'b': (2, 5), 'c': (3, 6)}

In [18]: dict(zip(list1, map(list, zip(map(int, list2), map(int, list3)))))
Out[18]: {'a': [1, 4], 'b': [2, 5], 'c': [3, 6]}

对于任意数量的列表,您可以这样做:

dict(zip(list1, zip(*(map(int, lst) for lst in (list2, list3, list4, ...)))))

或者,要将值列表变为元组而不是元组,

dict(zip(list1, map(list, zip(*(map(int, lst) for lst in (list2, list3, list4, ...))))))

1
像这样的东西是否可以修改,以便它适用于n个列表? - user1919
3
我喜欢dict(zip(keys,zip(values)))这个版本,特别是在Python3.x中,zip是惰性的。 - mgilson

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接