Python中的groupby分组功能的实例代码

当需要统计数据时，有时需要对数据进行分组操作。Python中提供了一个非常方便的分组工具——groupby函数。下面我们就来介绍一下groupby函数的使用。

什么是groupby函数？

groupby函数是Python内置的一个分组函数，它能够在对数据进行操作时，根据指定的键（key）对数据进行分组，并将每个分组的数据进行停留处理。

groupby函数的语法

groupby函数的语法为：

itertools.groupby(iterable, key=None)

其中，参数iterable是要进行分组的迭代器，key是用于分组的函数。当key被设置为None时，分组机制将使用原始的元素比较。

groupby函数的使用方法

下面我们以一个实例来说明groupby的使用方法：

我们需要统计一个班级中每个学生分数的平均值，并将结果按照分数段进行分组，最后求出每个分数段的平均分。假设班级中共有10个学生，我们先生成一个随机的成绩列表，如下所示：

import random

scores = [random.randint(0, 100) for _ in range(10)]
print(scores)

接下来我们使用groupby函数进行分组。由于我们要根据分数的大小进行分组，因此这里的key函数应该返回分数所在的分段。代码如下：

import itertools

def group_key(score):
    """分组键函数"""
    if score < 60:
        return "不及格"
    elif score < 70:
        return "及格"
    elif score < 80:
        return "中等"
    elif score < 90:
        return "优秀"
    else:
        return "卓越"

result = {}
for k, g in itertools.groupby(sorted(scores), group_key):
    result[k] = sum(g) / len(list(g))

print(result)

运行结果如下：

{'及格': 63.5, '中等': 74.66666666666667, '不及格': 30.333333333333332, '卓越': 95.0}

可以看出，我们成功地分组并统计了每个分数段的平均分。

groupby函数注意事项

• 在使用groupby函数时，首先必须对要分组的元素进行排序，否则会出现错误的分组结果；
• groupby函数返回的每个g元素是一个迭代器，需要将其转换为列表等数据类型，才能进行使用。