将嵌套列表转换为dict,其中list的第一个元素是dict的键

时间:2015-05-30 06:16:22

标签: python python-2.7

我有清单,

[['1', '4.00', 'A'],
['1', '5.00', 'B'],
['2', '4.00', 'V'],
['1', '12.00', 'C'],
['3', '4.00', 'R']]

如何将此列表转换为

{'1': [{'total': 21}, {'data': [['A'], ['B'], ['C']]}]}
{'2': [{'total': 4}, {'data': ['C']}]}
{'3':....}

1 个答案:

答案 0 :(得分:2)

首先,我们对基于初始元素对嵌套列表进行排序的数据进行排序,因此我们得到嵌套列表,其中元素在嵌套列表中的列表的第一个索引上排序,然后我们只需groupby()他们在第一个元素上。

import itertools

a= [['1', '4.00', 'A'],
   ['1', '5.00', 'B'],
   ['2', '4.00', 'V'],
   ['1', '12.00', 'C'],
   ['3', '4.00', 'R'],
]

a.sort()

dictionary = {}
group = itertools.groupby(a, key = lambda x:x[0])
for k,g in group:
   ans = 0
   alphabets = []
   for i in g:
      ans+=(float(i[1]))
      alphabets.append([i[2]])
   dictionary[k] = [{'total':ans}, {'date':alphabets}]

print dictionary

>>> {'1': [{'total': 21.0}, {'date': [['C'], ['A'], ['B']]}], '3': [{'total': 4.0}, {'date': [['R']]}], '2': [{'total': 4.0}, {'date': [['V']]}]}