itertools中的groupby可对可迭代对象进行分组。
形式为:
itertools.groupby(iterable, key=None)
参数:
iterable: 一个可迭代对象,对其进行分组。
key: 分组用的函数,该函数作用于iterable的每个元素,若是返回相同的值就认为是同一组。
返回:返回由元组组成的列表。元组的第一个值是key返回的值,第二个值是一个可迭代对象,就是该分组的可迭代对象。
应用:
一般需要先对iterable进行一定的排序,再做groupby
例子:
from itertools import groupby
l = [{'a': 1, 'b': 1}, {'a': 2, 'b': 2}, {'a': 1, 'b': 3}]
l = sorted(l, lambda p: p['a'])
for name, group in groupby(l, lambda: p: p['a']):
print(name)
pring(group)
结果:
1
<itertools._grouper object at 0x7f77befc1ba8>
2
<itertools._grouper object at 0x7f77c27ed470>
如果不进行sorted,返回的结果会是:
1
<itertools._grouper object at 0x7f77befd9320>
2
<itertools._grouper object at 0x7f77c27ed710>
1
<itertools._grouper object at 0x7f77c27ed470>