天天看点

python groupby用法_Python 标准库实践之合并字典组成的列表

python groupby用法_Python 标准库实践之合并字典组成的列表

作者 | 陈祥安 来源 | Python学习开发

python groupby用法_Python 标准库实践之合并字典组成的列表

最近做了个数据合并的需求这里记录下:

要求如下:

dict_of_list1 = [{"name": "zhangsan", "age": 18}, {"name": "lisi", "age": 19}, {"name": "wangwu", "age": 20}]dict_of_list2 = [{"name": "zhangsan", "province": "shandong"}, {"name": "lisi",                                                                "province": "changsha"},                 {"name": "xiaohong", "province": "guangxi"}, {"name": "wangwu",                                                               "province": "xizang"}]
           

将上面的两个列表的字典进行合并将名字一样的项目放在一起。

形如:

[{"name":"zhangsan","age":18,"province":"shandong"},{"name":"xiaohong","province":"guangxi"}]
           

传统解法

dict_of_list1 = [{"name": "zhangsan", "age": 18}, {"name": "lisi", "age": 19}, {"name": "wangwu", "age": 20}]dict_of_list2 = [{"name": "zhangsan", "province": "shandong"}, {"name": "lisi",                                                                "province": "changsha"},                 {"name": "xiaohong", "province": "guangxi"}, {"name": "wangwu",                                                               "province": "xizang"}]all_data = dict_of_list1 + dict_of_list2d = {}for item in all_data:    name = item["name"]     if name in d:         d[name].update(item)    else:        d[name] = itemresult = []for k,v in d.items():    result.append(v)sorted(v)print(result)
           

根据要求得知,我们需要根据姓名一样的进行合并,那么可以创建一个新的字典可以把姓名当作字典的键,然后判断这个键是否在这个字典中,如果不存在就是赋值,否则去更新其值。因为它的值是字典类型所以可以通过update进行更新,这种方法可以适用到其他语言。

下面再来一个python独有的解法。

通过分组函数

利用python标准库

from operator import itemgetterfrom itertools import groupbyfrom collections import ChainMapdict_of_list1 = [{"name": "zhangsan", "age": 18}, {"name": "lisi", "age": 19}, {"name": "wangwu", "age": 20}]dict_of_list2 = [{"name": "zhangsan", "province": "shandong"}, {"name": "lisi",                                                                "province": "changsha"},                 {"name": "xiaohong", "province": "guangxi"}, {"name": "wangwu",                                                               "province": "xizang"}]key = itemgetter("name")all_data = dict_of_list1 + dict_of_list2all_data.sort(key=key)result_list = []for x, y in groupby(all_data, key=key):    d = dict(ChainMap(*y))    result_list.append(d)sorted(result_list, key=key)print(result_list)
           

我们可以使用itertools模块的groupby按照name进行分组

groupby有两个参数第一个是一个可迭代对象,第二个是指定按照什么去分组,类似我们排序时指定key的值,

这里key的值可以使用lambda表达式,或者使用标准库operator的itemgetter方法,实现相同的效果。

使用这个方法有一个要求那就是,需要提前对可迭代对象进行排序。

经过groupby修饰之后我们得到一个key和

itertools._grouper

对象,key就是我们指定那个键的值,这里就是name的值,然后可以对

itertools._grouper

对象进行拆分,得到几个字典,再通过collections的ChainMap对

其元素进行字典合并,之后转为dict对象。最终加入到指定列表完成我们的任务。

该方法主要是熟悉Python一些比较实用的标准库的用法。

···END···

精彩推荐

python groupby用法_Python 标准库实践之合并字典组成的列表

 互联网人办公地鄙视链

python groupby用法_Python 标准库实践之合并字典组成的列表

 程序员的双肩包,大概能装下整个宇宙!

python groupby用法_Python 标准库实践之合并字典组成的列表

 前端学习资源分享,含Vue3.0全家桶,赠知识图谱!

python groupby用法_Python 标准库实践之合并字典组成的列表

 简单!粗暴!这10门「热门技术」课程,直接免费看!

python groupby用法_Python 标准库实践之合并字典组成的列表
python groupby用法_Python 标准库实践之合并字典组成的列表