首页 > 后端开发 > Python教程 > 如何识别 Python 列表中的连续数字并对其进行分组?

如何识别 Python 列表中的连续数字并对其进行分组?

Susan Sarandon
发布: 2024-12-08 17:47:11
原创
711 人浏览过

How to Identify and Group Consecutive Numbers in a Python List?

识别列表中的连续数字

任务是将列表分为连续数字组。每组应仅包含连续的数字。输出必须保留单个数字,而不是将它们组合成范围。

解决方案:

在 Python 中,您可以使用内置的 groupby 函数和自定义键达到this:

from itertools import groupby
from operator import itemgetter

ranges = []
for key, group in groupby(enumerate(data), lambda (index, item): index - item):
    group = map(itemgetter(1), group)
    if len(group) > 1:
        ranges.append(xrange(group[0], group[-1]))
    else:
        ranges.append(group[0])
登录后复制

解释:

  1. enumerate(data) 创建一个对列表,其中每个对由一个索引(从 0 开始)以及 data 中对应的元素。
  2. lambda 函数 (lambda (index, item): index - item) 计算索引和元素值之间的差异。这种差异作为分组的关键。
  3. groupby 根据该键将对列表划分为连续的组。每个组都包含具有相同差异的元素。
  4. 为了检索连续数字的范围,我们使用 map(itemgetter(1), group) 将每个组中的对映射到其第二个元素(实际数字)。
  5. 我们检查每一组是否包含多个数字。如果是这样,我们为这些数字创建一个范围并将其附加到范围中。否则,我们只需将数字添加到范围中即可。

示例输出:

data = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20]
ranges = [xrange(2, 5), xrange(12, 17), 20]
登录后复制

以上是如何识别 Python 列表中的连续数字并对其进行分组?的详细内容。更多信息请关注PHP中文网其他相关文章!

本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板