识别列表中的连续数字组是各种编程场景中出现的常见任务。
幸运的是,Python 的内置函数和模块提供了优雅的这个问题的解决方案。一种方法利用 groupby() 函数和 lambda 表达式:
from itertools import groupby data = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17] for k, g in groupby(enumerate(data), lambda (i, x): i - x): print(map(itemgetter(1), g))
此代码段根据索引和值之间的差异对连续数字进行分组,确保每个组中的元素是连续的。
如果想要的输出是表示连续数字范围的元组列表,可以进行以下修改:
ranges = [] for k, g in groupby(enumerate(data), lambda (i, x): i - x): group = map(itemgetter(1), g) ranges.append((group[0], group[-1]))
或者,可以使用循环和列表理解来实现自定义解决方案:
ranges = [] for i in range(len(data)): if i == len(data) - 1 or data[i] + 1 != data[i + 1]: ranges.append((data[i], data[i])) else: start = i while i + 1 < len(data) and data[i] + 1 == data[i + 1]: i += 1 ranges.append((data[start], data[i]))
此方法迭代列表,识别每个连续组的开始和结束索引,并将它们附加到范围列表中。
以上是如何识别 Python 列表中的连续数字组?的详细内容。更多信息请关注PHP中文网其他相关文章!