[英]How do you create a dictionary from nested lists in Python?
我希望可能会有某种方式使用理解来做到这一点,但是我说我的数据看起来像这样:
data = [['a', 'b', 'c'], [1, 2, 3], [4, 5, 6]]
我的最终目标是创建一个字典,其中第一个嵌套列表保存键,其余列表保存值:
{'a': [1, 4], 'b': [2, 5], 'c': [3, 6]}
我已经尝试过类似的方法使我接近,但是可以看出我在将列表添加到字典值时遇到了麻烦,并且此代码只是覆盖了:
d = {data[0][c]: [] + [col] for r, row in enumerate(data) for c, col in enumerate(row)}
>>> d
{'c': [6], 'a': [4], 'b': [5]}
您可以在dict理解中使用zip:
{z[0]: list(z[1:3]) for z in zip(*data)}
Out[16]: {'a': [1, 4], 'b': [2, 5], 'c': [3, 6]}
zip将采取转置 :
list(zip(['a', 'b', 'c'], [1, 2, 3], [4, 5, 6]))
Out[19]: [('a', 1, 4), ('b', 2, 5), ('c', 3, 6)]
但是,您的数据是列表的列表,因此为了确保Python不会看到一个列表,而是看到三个单独的列表,您需要zip(*data)
而不是zip(data)
。 关于*
的使用,有几篇文章: (1) , (2) , (3) 。
list(zip(*data))
Out[13]: [('a', 1, 4), ('b', 2, 5), ('c', 3, 6)]
在dict理解中,您将第一个元素作为键,将其余两个作为值。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.