Python itertools.groupby函数的应用场景及使用方法

小蓝 • 2024-12-15 12:46 • 编程

一、groupby函数概述

groupby函数可以将一个序列按照指定的标准进行分组，返回一个根据分组的标准而创建的分组对象，可以通过循环迭代获取分组和分组中的元素。groupby函数是itertools模块的函数。

groupby函数的使用方法如下：


import itertools

key_func = lambda x: x[0]
grouped_data = itertools.groupby(iterable, key_func)
for key, group in grouped_data:
    for item in group:
        # do something with item

其中，iterable是要分组的序列，可以是列表、元组、生成器等可迭代对象；key_func是指定分组的标准，它是一个函数，接收一个元素作为参数，返回一个表示分组标准的值；grouped_data是分组后的对象，它包含了分组后的每个子集和对应的键。

二、groupby函数的应用场景

1.分组统计

groupby函数可以用于分组统计数据。假设我们有一个列表存储了每个人的姓名、年龄和性别，我们可以使用groupby函数按照性别对其进行分组，并统计每个性别的人数和平均年龄。


import itertools

people = [('Alice', 25, 'F'), ('Bob', 30, 'M'), ('Charlie', 20, 'M'), ('David', 27, 'M'), ('Eve', 22, 'F')]

# 按照性别进行分组，并统计每个性别的人数和平均年龄
gender_func = lambda x: x[2]
grouped_data = itertools.groupby(sorted(people, key=gender_func), key_func=gender_func)
for gender, group in grouped_data:
    count = 0
    age_sum = 0
    for person in group:
        count += 1
        age_sum += person[1]
    avg_age = age_sum / count
    print(f"Gender: {gender}, Count: {count}, Average Age: {avg_age}")

输出结果为：


Gender: F, Count: 2, Average Age: 23.5
Gender: M, Count: 3, Average Age: 25.666666666666668

2.去除连续重复元素

groupby函数可以用于去除连续重复的元素。假设我们有一个列表，其中包含了连续的重复元素，我们可以使用groupby函数将其去重。


import itertools

data = [1, 1, 2, 2, 3, 3, 3, 4, 4, 5]

# 去除连续重复元素
grouped_data = itertools.groupby(data)
result = [k for k, g in grouped_data]
print(result)  # [1, 2, 3, 4, 5]

3.按照指定长度分组

groupby函数可以用于按照指定长度对序列进行分组。假设我们有一个列表，我们想将其按照行切分成若干个小列表，我们可以使用groupby函数实现。


import itertools

data = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 按照长度为3进行分组
grouped_data = itertools.groupby(enumerate(data), key=lambda x: x[0] // 3)
result = [[x[1] for x in g] for k, g in grouped_data]
print(result)  # [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

三、小结

groupby函数是一个非常强大的分组函数，可以应用于各种不同的场景。我们可以使用它进行分组统计、去重、按照指定长度分组等操作。在使用groupby函数时需要注意分组标准的设置，以及获取分组结果的方式。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/257664.html