python - 查找列表中连续数字的开头和结尾

Question

我有一个输入列表:

s = [1, 2, 3, 5, 6, 9, 10, 11]

我正在尝试将以下内容作为输出:

lows = [1, 5, 9]
highs = [3, 6, 11]

lows 列表包含列表中连续数字的所有子列表的所有最小数字:[1, 2, 3]、[5, 6] 和 [9, 10, 11]。高点反之。

• 一个子列表保证至少有 2 个成员。
• 输入列表中的所有数字都是唯一的。

完成此任务的最简洁方法是什么？我想知道 itertools 中是否有什么东西可以很好地做到这一点？

Answer 1

>>> from itertools import groupby, count
>>> from operator import itemgetter
>>> s = [1, 2, 3, 5, 6, 9, 10, 11]
>>> ig = itemgetter(0, -1)
>>> lows, highs = zip(*(ig(list(g)) for k, g in groupby(s, lambda x, c=count(): x-next(c))))
>>> lows
(1, 5, 9)
>>> highs
(3, 6, 11)