Question

我有一个输入列表：

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

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

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

lows列表包含列表中连续数字的所有子列表中的所有最低数量：[1, 2, 3], [5, 6], and [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)

Answer 2

总有明显的手动方法：

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

def runs(iterable):
    iterator = iter(iterable)
    start = end = next(iterator)

    for item in iterator:
        if item != end + 1:
            yield (start, end)
            start = item

        end = item

    yield (start, end)

lows, highs = zip(*runs(s))

lows
#>>> (1, 5, 9)

highs
#>>> (3, 6, 11)

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

2 个答案: