某公司的面试题

最近面试一个公司的面试题,对数据的格式进行一个转换

## 输入格式
input_list = [
    {'a':'tata', 'b': 'foo'},
    {'a':'pipo', 'b': 'titi'},
    {'a':'pipo', 'b': 'toto'},
    {'a':'tata', 'b': 'bar'}
]  
## 输出格式
output_dict = {
    'pipo': [
        {'a': 'pipo', 'b': 'titi'},
        {'a': 'pipo', 'b': 'toto'}
    ],
    'tata': [
        {'a': 'tata', 'b': 'foo'},
        {'a': 'tata', 'b': 'bar'}
    ]
}    

代码实现如下:

tmp_set=set()
output_dict={}
for i in input_list:
    tmp_set.add(i.get("a"))
for key in tmp_set:
    tmp_list=[]
    for list_temp in input_list:
        if key in list_temp.values():
            tmp_list.append(list_temp)
    output_dict[key]=tmp_list

 

输出结果如下:

 

 

  

posted @ 2021-04-04 16:37  一切都是当下  阅读(58)  评论(0编辑  收藏  举报