我希望可能会有某种方式使用一种理解来做到这一点,但是我说我的数据看起来像这样:
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理解中,您将第一个元素作为键,将其余两个作为值。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句