如何高效识别Python列表中的连续数字组？

识别列表中的连续数字组

当面临识别列表中连续数字组的任务时，有多种方法需要考虑。

内置 Python功能

Python 为这项任务提供了一个强大的工具：groupby() 函数。利用这个函数，我们可以根据元素的索引和值之间的差异对元素进行分类。例如，在列表 [2, 3, 4, 5, 12, 13, 14, 15, 16, 17] 中，groupby() 函数将输出以下组：

[(2, [2, 3, 4, 5]), (12, [12, 13, 14, 15, 16, 17])]

这里，每组由索引与其值相差相同数量的元素组成，表示连续的数字。

自定义函数

另一种方法是创建自定义函数。下面给出了一个实现：

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])

该函数生成一个元组列表，其中每个元组代表一个连续的数字组。对于输入 [2, 3, 4, 5, 12, 13, 14, 15, 16, 17]，输出将为：

[(2, 5), (12, 17), 20]

请注意，单个数字作为单个元素返回，而不是范围。

以上是如何高效识别Python列表中的连续数字组？的详细内容。更多信息请关注PHP中文网其他相关文章！