python itertools groubpy 实例

最新推荐文章于 2024-03-11 14:26:34 发布

mars914

最新推荐文章于 2024-03-11 14:26:34 发布

阅读量310

点赞数 2

分类专栏： python 文章标签： python groupby

本文链接：https://blog.csdn.net/mars914/article/details/84509439

版权

python 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

本文介绍了Python中itertools模块的groupby函数用法，通过两个示例展示了如何根据列表中的特定元素对数据进行分组。第一个示例根据日期对记录进行分组；第二个示例则进一步根据日期和名称进行更详细的分组。

摘要由CSDN通过智能技术生成

groupby(iterable [,key]):

创建一个迭代器，对iterable生成的连续项进行分组，在分组过程中会查找重复项。

如果iterable在多次连续迭代中生成了同一项，则会定义一个组，如果将此函数应用一个分类列表，那么分组将定义该列表中的所有唯一项，key（如果已提供）是一个函数，应用于每一项，如果此函数存在返回值，该值将用于后续项而不是该项本身进行比较，此函数返回的迭代器生成元素(key, group)，其中key是分组的键值，group是迭代器，生成组成该组的所有项。

an example:

from itertools import groupby
from operator import itemgetter

things =[('2009-09-02', 'mars', 1, 11),
          ('2009-09-02', 'mars', 1, 3),
          ('2009-09-03', 'ma', 2, 10),
          ('2009-09-03', 'ma', 2, 4),
          ('2009-09-03', 'ma', 2, 22),
          ('2009-09-06', 'm', 3, 33)]

for key, items in groupby(things, itemgetter(0)):
    print key
    for subitem in items:
        print subitem
    print '-' * 20

The output looks like:

2009-09-02
('2009-09-02', 'mars, 11)
('2009-09-02', 'mars', 3)
--------------------
2009-09-03
('2009-09-03', 'ma', 10)
('2009-09-03', 'ma', 4)
('2009-09-03', 'ma', 22)
--------------------
2009-09-06
('2009-09-06', 'm', 33)
-------------------

another example:

key_func = lambda t: (t[0], t[1])
for key, items in groupby(things, key_func) :
    print key
    for subitem in items:
        print subitem
    print '-' * 20

 output:

('2009-09-02', 'mars')
('2009-09-02', 'mars, 11)
('2009-09-02', 'mars', 3)
--------------------
('2009-09-03', 'ma')
('2009-09-03', 'ma', 10)
('2009-09-03', 'ma', 4)
('2009-09-03', 'ma', 22)
--------------------
('2009-09-06', 'm')
('2009-09-06', 'm', 33)
-------------------