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

识别列表中的连续数字

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

解决方案：

在 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中文网其他相关文章！