groupby(iterable[, keyfunc])
-> create an iterator which returns
(key, sub-iterator) grouped by each value of key(value).
將key函數(shù)作用于序列的各個(gè)元素江解。根據(jù)key函數(shù)結(jié)果，將擁有相同函數(shù)結(jié)果的元素分到一個(gè)新的迭代器奸柬。每個(gè)新的迭代器以函數(shù)返回結(jié)果為標(biāo)簽偎捎。
注意：分組之前需要使用sorted()對(duì)原循環(huán)器的元素，根據(jù)key函數(shù)進(jìn)行排序讳癌，讓同組元素先在位置上靠攏穿稳。
示例：
from operator import itemgetter
from itertools import groupby
data = [
{'name': 'zhangsan', 'age': 20, 'country': 'China'},
{'name': 'lisi', 'age': 19, 'country': 'USA'},
{'name': 'wangwu', 'age': 22, 'country': 'JP'},
{'name': 'zhaoliu', 'age': 21, 'country': 'USA'},
{'name': 'maqi', 'age': 22, 'country': 'USA'},
{'name': 'yangba', 'age': 18, 'country': 'China'}
]
# 通過country進(jìn)行分組
data.sort(key=itemgetter('country')) # 需要先排序，然后才能groupby
lstg = groupby(data, itemgetter('country'))
for key, group in lstg:
for g in group: # group是一個(gè)迭代器晌坤，包含了所有的分組列表
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'}
"""
# 返回每個(gè)分組的個(gè)數(shù)
res = dict([(key, len(list(group))) for key, group in lstg])
"""
{'JP': 1, 'China': 2, 'USA': 3}
"""
自定義分組
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])]
"""
