itertools模块之groupby

最新推荐文章于 2024-03-04 13:45:50 发布

不爱打代码的程序员

最新推荐文章于 2024-03-04 13:45:50 发布

阅读量450

点赞数

CC 4.0 BY-SA版权

文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_40109570/article/details/127365578

这个博客展示了如何利用Python的itertools.groupby函数，结合自定义函数gb，将数字列表按特定条件分组。通过排序和gb函数，将数字分为'less'(小于等于10)、'middle'(11到29之间)和'great'(大于等于30)三个类别，并以字典形式展示分组结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

itertools模块之groupby

from itertools import groupby

lst = [2, 8, 11, 25, 43, 6, 9, 29, 51, 66]


def gb(num):
    if num <= 10:
        return 'less'
    elif num >= 30:
        return 'great'
    else:
        return 'middle'


print({k:list(g) for k, g in groupby(sorted(lst), key=gb)})
#{'less': [2, 6, 8, 9], 'middle': [11, 25, 29], 'great': [43, 51, 66]}