注册 | 登录

张浩然的个人站



用groupby对可迭代对象进行分组

itertools中的groupby可对可迭代对象进行分组。

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>

  Python

请点击登录发表评论


About ME

about me

张浩然

朝三暮四的的执着青年,嵌入式打杂工程师,web全沾工程师

伪文青,真宅男,历史爱好者

TAG cloud