用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>
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值