Python中的分组函数(groupby、itertools)
2017-09-28 14:23
567 查看
from operator import itemgetter #itemgetter用来去dict中的key,省去了使用lambda函数 from itertools import groupby #itertool还包含有其他很多函数,比如将多个list联合起来。。 d1={'name':'zhangsan','age':20,'country':'China'} d2={'name':'wangwu','age':19,'country':'USA'} d3={'name':'lisi','age':22,'country':'JP'} d4={'name':'zhaoliu','age':22,'country':'USA'} d5={'name':'pengqi','age':22,'country':'USA'} d6={'name':'lijiu','age':22,'country':'China'} lst=[d1,d2,d3,d4,d5,d6] #通过country进行分组: lst.sort(key=itemgetter('country')) #需要先排序,然后才能groupby。lst排序后自身被改变 lstg = groupby(lst,itemgetter('country')) #lstg = groupby(lst,key=lambda x:x['country']) 等同于使用itemgetter() for key,group in lstg: for g in group: #group是一个迭代器,包含了所有的分组列表 print key,g 返回: China {'country': 'China', 'age': 20, 'name': 'zhangsan'} China {'country': 'China', 'age': 22, 'name': 'lijiu'} JP {'country': 'JP', 'age': 22, 'name': 'lisi'} USA {'country': 'USA', 'age': 19, 'name': 'wangwu'} USA {'country': 'USA', 'age': 22, 'name': 'zhaoliu'} USA {'country': 'USA', 'age': 22, 'name': 'pengqi'} print [key for key,group in lstg] #返回:['China', 'JP', 'USA'] print [(key,list(group)) for key,group in lstg] #返回的list中包含着三个元组: [('China', [{'country': 'China', 'age': 20, 'name': 'zhangsan'}, {'country': 'China', 'age': 22, 'name': 'lijiu'}]), ('JP', [{'country': 'JP', 'age': 22, 'name': 'lisi'}]), ('USA', [{'country': 'USA', 'age': 19, 'name': 'wangwu'}, {'country': 'USA', 'age': 22, 'name': 'zhaoliu'}, {'country': 'USA', 'age': 22, 'name': 'pengqi'}])] print dict([(key,list(group)) for key,group in lstg]) #返回的是一个字典: {'JP': [{'country': 'JP', 'age': 22, 'name': 'lisi'}], 'China': [{'country': 'China', 'age': 20, 'name': 'zhangsan'}, {'country': 'China', 'age': 22, 'name': 'lijiu'}], 'USA': [{'country': 'USA', 'age': 19, 'name': 'wangwu'}, {'country': 'USA', 'age': 22, 'name': 'zhaoliu'}, {'country': 'USA', 'age': 22, 'name': 'pengqi'}]} print dict([(key,len(list(group))) for key,group in lstg]) #返回每个分组的个数: {'JP': 1, 'China': 2, 'USA': 3}
#返回包含有2个以上元素的分组 print [key for key,group in groupby(sorted(lst,key=itemgetter('country')),itemgetter('country')) if len(list(group))>=2] #返回:['China', 'USA']
lstg = groupby(sorted(lst,key=itemgetter('country')),key=itemgetter('country')) lstgall=[(key,list(group)) for key,group in lstg ] print dict(filter(lambda x:len(x[1])>2,lstgall)) #过滤出分组后的元素个数大于2个的分组,返回: {'USA': [{'country': 'USA', 'age': 19, 'name': 'wangwu'}, {'country': 'USA', 'age': 22, 'name': 'zhaoliu'}, {'country': 'USA', 'age': 22, 'name': 'pengqi'}]}
自定义分组:
from itertools import groupby lst=[2,8,11,25,43,6,9,29,51,66] def gb(num): if num <= 10: return 'less' elif num >=30: return 'great' else: return 'middle' print [(k,list(g))for k,g in groupby(sorted(lst),key=gb)] 返回: [('less', [2, 6, 8, 9]), ('middle', [11, 25, 29]), ('great', [43, 51, 66])]
相关文章推荐
- Python中的分组函数(groupby、itertools)
- Python中的分组函数(groupby、itertools)
- 详解Python中的分组函数groupby和itertools)
- [PY3]——根据某个特定的字段来分组迭代访问一个字段或序列?/ itertools.groupby( )函数
- python itertools的函数(含有排列组合函数)
- [转]可爱的 Python:: 使用 itertools 模块中的组合函数
- python中itertools模块zip_longest函数详解
- Python:itertools模块-groupby
- python中Groupby使用(四)-通过函数进行分组
- pandas中的groupby函数的分组结果怎么保存成DataFrame
- Python中itertools模块用法详解
- Python itertools 模块
- Python标准库13 循环器 (itertools)学习
- python itertools模块
- 【Python模块】functools —— 作用于可调用对象的高阶函数集合
- pandas中的groupby函数的分组结果怎么保存成DataFrame
- python内建模块之itertools
- python中itertools包无穷循环器的使用
- Python迭代器包itertools(转)
- Python模块 - itertools循环器模块